用Tensorboard在VScode Remote ssh中显示图像

可以用Tensorboard在vscode的ssh连接中显示plot出的图像. 具体方法如下: from torch.utils.tensorboard import SummaryWriter writer = SummaryWriter(log_dir='./logdir/') fig = pl
posted @ 2024-05-01 14:31  酱_油  阅读(32)  评论(0编辑  收藏  举报