tensorboardX是PyTorch的tensorboard插件,支持模型图可视化

tensorboardX是PyTorch的tensorboard插件,支持模型图可视化