matplotlib画图通过tensorboardX显示(pytorch)
问题描述
由于远程服务器上无法显示实时生成的图片,通常比较笨的方法是通过保存下来改图片,然后下载到本地,这个方法费时又费力。因此,这里我决定介绍一种通过tosorboardX的方法来显示图像。
具体步骤
如果是在docker上首先需要对tensorboard的6006端口进行映射。
对应示例代码
import torch
from torch.autograd import Variable
...
原创
2019-03-24 21:08:38 ·
2440 阅读 ·
0 评论