关于tensorboard上没有数据问题
使用tensorboard绘制Loss曲线时,发现数据总是无法上传,在tensorboard可视化界面里始终显示”无数据“
原因最后发现是:
writer = SummaryWriter("./logs")
这行代码写在了if __name__ == 'main'
函数之外,而没有写在函数内,这会造成logs文件夹内多出了很多文件。
目前原因不太清楚,但似乎是pytorch中的dataloader影响了:
- 如果writer写在main函数外时,在运行到dataloader时,发现logs中多出了很多文件
- writer写在main函数内时,在运行到dataloader时,logs中始终只有一个文件
此时tensorboard可视化界面就可以画图了