深度学习多GPU训练时，tensorboard无法显示loss曲线图的问题。

在进行双卡训练一个神经网络时，我用DistributedDataParallel的方法进行分布式训练，但是发现生成的loss曲线只有第一个点。解决方法是在这之前添加args.local_rank==0的约束，和多卡保存ckpt的思路一样。深度学习多GPU训练时，tensorboard无法显示scalar曲线图的问题。

初秦

758人浏览 · 2023-03-07 14:59:35

初秦 · 2023-03-07 14:59:35 发布

问题描述

深度学习多GPU训练时，tensorboard无法显示scalar曲线图的问题。
在进行双卡训练一个神经网络时，我用DistributedDataParallel的方法进行分布式训练，但是发现生成的loss曲线只有第一个点。
解决方法是在这之前添加args.local_rank==0的约束，和多卡保存ckpt的思路一样。如下：

if args.local_rank==0
	summary_writer.add_scalar(name,value,global_step)
	...

魔乐社区

魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。

更多推荐

小参数・大码力・易部署 | Qwen3.6-27B上线魔乐社区，基于昇腾的部署教程来了

继一周前模型开源发布后，千问再度开源Qwen3.6-27B —— 一个拥有270亿参数的稠密多模态模型，也是社区呼声最高的模型规格。Qwen3.6-27B 依然支持多模态思考与非思考模式，在智能体编程方面达到了旗舰级表现，全面超越前代开源旗舰 Qwen3.5-397B-A17B（总参数397B / 激活参数17B的MoE模型）。作为稠密架构，它无需MoE路由即可部署，是开发者在实用、可广泛部署规模