admin管理员组

文章数量:1635979

如果你在谷歌云端的colab上用pytorch框架打开tensorboard,发生以下打不开的情况,其实只需要改动一点点代码

原代码(点击网站无法显示):
我是想将logs函数显示出来,范围是x轴100以内

from torch.utils.tensorboard import SummaryWriter
writer=SummaryWriter('logs')
for i in range(100):
  writer.add_scalar("y=x",i,i)
!tensorboard --logdir=logs --port=6006   #可以指定端口,也可以不指定

以下是可以出来结果的执行代码,分为两步

1、先安装tensorboardX包

! pip install tensorboardX
%load_ext tensorboard

2、将上文中的!改为%,这样不用点击网页,而是直接显示在运行结果处

from torch.utils.tensorboard import SummaryWriter
writer=SummaryWriter('logs')
for i in range(100):
  writer.add_scalar("y=x",i,i)
%tensorboard --logdir=logs --port=6006   # 改动的地方,将!改为%

结果运行如下:

本文标签: 解决办法打不开TensorBoardColab