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