Declare a backend for matplotlib (#171)
Declare a backend for matplotlib (#171)

This demo doesn't work on my Linux machine because it doesn't declare a backend for matplotlib. With this change it can now work on my server. This seems to be a common issue as described here: https://github.com/matplotlib/matplotlib/issues/3466/#issuecomment-195899517
lanpa/tensorboardX,lanpa/tensorboardX