在服务器上远程使用tensorboard查看训练loss和准确率
生活随笔
收集整理的這篇文章主要介紹了
在服务器上远程使用tensorboard查看训练loss和准确率
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
本人使用的是vscode 很簡單
from torch.utils.tensorboard import SummaryWriter writer = SummaryWriter('./logs')writer.add_scalar('train_loss',loss.val(),iteration) # 名字,數據,迭代次數訓練的過程中會產生一個./logs的文件夾,里面存放的是保存的訓練loss數據,
然后在命令行運行
tensorboard --logdir './logs'總結
以上是生活随笔為你收集整理的在服务器上远程使用tensorboard查看训练loss和准确率的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: MyBatis的逆向工程工具,自动生成数
- 下一篇: python123第五章_python