AutoDL中使用TensorBoard

# 示例代码
log_dir = 'logs'
writer = SummaryWriter(log_dir=log_dir)
# 使用 TensorBoard 记录:
writer.add_scalar('Loss/train', train_loss, epoch)

 此时会把模型存储在“项目文件/logs”中,

1.如果是在本地训练的模型,那么在终端输入

activate 环境名称
tensorboard --logdir=logs

点击终端打印的链接,即可查看tensorboard页面。

2.如果是在autodl服务器上训练模型,应当使用以下方法:

# 在终端中执行以下命令启动TensorBoard
# 删除当前tensorboard进程
ps -ef | grep tensorboard | awk '{print $2}' | xargs kill -9    
# 重新启动
tensorboard --port 6007 --logdir 【/指定目录】

注意:代码在服务器中默认的目录是“/tmp/tmp目录下自己的项目名称”,

所以【/指定目录】应当为(以作者本人的项目为例): 

tensorboard --port 6007 --logdir /tmp/pycharm_project_149/logs

然后进入AutoPanel,即可查看tensorboard。

官方文档:https://www.autodl.com/docs/tensorboard/

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值