tensorboardX 1.7 is bugged

[shahbuland]

You should change requirements and readme to show tensorboardX==1.6 is required since 1.7 seems to be bugged (read another issue on different repo saying it's broken) and train.py will not run with it