用Tensorboard在VScode Remote ssh中显示图像

可以用Tensorboard在vscode的ssh连接中显示plot出的图像.

 

具体方法如下:

 

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter(log_dir='./logdir/')


fig = plt.figure()
# 绘图操作
writer.add_figure('figname', fig, 0)

 

 

在ssh中:   tensorboard --logdir ./

 

在浏览器中打开 6006 端口即可看到图像