【深度学习训练使用tensorboard】

pycharm+服务器+pytorch+tensorboard:

查看train test loss:

主要包括pycharm里的代码部分和ssh里的操作


代码部分

提示:这里描述项目中遇到的问题:

例如:数据传输过程中数据不时出现丢失的情况,偶尔会丢失一部分数据
APP 中接收数据代码:

from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter('/data/xxxx/xxxx/xxxx/xxxx/runs')
#在每个epoch的操作如下
for epoch in range(start_epoch, epochs):
	Loss_list = 0
	train_epoch(Loss_list,xxxx))
	#train函数,里面会把每个patch的总loss求出来放在Loss_list
	writer.add_scalar(tag="loss/train", scalar_value=Loss_list,
                          global_step=epochs)
	Loss_list = 0
    print("\ntest")
    test_epoch(Loss_list,xxxx))
    writer.add_scalar(tag="loss/test", scalar_value=Loss_list,
                          global_step=epochs)	

在ssh里操作:

ssh在远程服务器上启动tensorboard,注意!要cd到存放tfevents文件夹的上一级文件夹位置,不然会出现**No dashboards are active for the current data set.**的问题。
在这里插入图片描述
然后在浏览器打开http://服务器IP:6066/
即可
tensorboard会读取服务器上项目里面的runs文件夹文件
在这里插入图片描述
参考
TensorBoard 可视化 不显示数据问题 No scalar data was found…
pytorch下tensorboard的使用

PS
1.画多条线用add_scalars
2.writer.close()之后,曲线才能显示完全
3.假设训练到100个epoch,如果要从100个开始继续训练,0-100的数据不会被覆盖掉,而是会从100个后面开始新增数据。同样的,如果说又重新从0-100训练,则原本的0-100个epoch的数据会被重写。
4.最后一个event file不要删掉即可,前面的没啥用,即使像3所说,从100开始继续训练,删掉前面的event file也不会吧0-100的数据删掉。

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
AutoDL(Automated Deep Learning)是一种自动化深度学习的方法,它旨在简化和加速深度学习模型的开发过程。TensorBoard是TensorFlow提供的一个可视化工具,用于分析和监控深度学习模型的训练过程和性能。 要使用TensorBoard来监控AutoDL的训练过程,可以按照以下步骤进行操作: 1. 安装TensorFlow和TensorBoard:首先,确保你已经安装了TensorFlow和TensorBoard。可以使用pip命令来安装它们: ``` pip install tensorflow pip install tensorboard ``` 2. 在AutoDL代码中添加TensorBoard回调:在你的AutoDL代码中,使用TensorFlow提供的`tf.keras.callbacks.TensorBoard`回调函数来将TensorBoard集成到训练过程中。在回调函数的构造函数中,可以指定TensorBoard日志文件的保存路径等参数。 3. 运行训练脚本:运行你的AutoDL训练脚本,确保在训练过程中启用了TensorBoard回调。训练过程中,TensorBoard会将训练指标和其他相关信息写入到指定的日志文件中。 4. 启动TensorBoard服务器:在训练完成后,打开终端并导航到保存日志文件的目录。然后运行以下命令启动TensorBoard服务器: ``` tensorboard --logdir=./logs ``` 这里的`./logs`是你保存日志文件的路径,根据实际情况进行修改。 5. 访问TensorBoard界面:在浏览器中输入`http://localhost:6006`,即可访问TensorBoard的界面。在界面上,你可以查看训练过程中的损失曲线、准确率等指标,还可以可视化模型的计算图、参数分布等信息。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值