pytorch框架使用tensorboard安装避坑
·
具体参考博客:https://blog.csdn.net/bigbennyguo/article/details/87956434
1.安装好了pytorch
2.安装tensorboard
pip install tensorboard
3.安装tensorboardX
pip install tensorboardX
4.测试成功
#Tensorboard的使用,图像变换transform的使用
from tensorboardX import SummaryWriter
writer=SummaryWriter("log")
for i in range(100):
writer.add_scalar("y=x",i,i)
writer.close()

更多推荐



所有评论(0)