如何解决使用 Pytorch Lightning 时如何将指标例如验证损失记录到 TensorBoard?
我使用 Pytorch Lightning 训练我的模型(在 GPU 设备上,使用 DDP),TensorBoard 是 Lightning 使用的默认记录器。
我的代码设置为分别记录每个训练和验证步骤的训练和验证损失。
class MyLightningModel(pl.LightningModule):
def training_step(self,batch):
x,labels = batch
out = self(x)
loss = F.mse_loss(out,labels)
self.log("train_loss",loss)
return loss
def validation_step(self,labels)
self.log("val_loss",loss)
return loss
TensorBoard 在 train_loss
选项卡中正确绘制了 val_loss
和 SCALERS
图表。但是,在 HPARAMS
标签的左侧栏上,只有 hp_metric
在 Metrics
下可见。
但是,在 HPARAMS
标签的左侧栏上,只有 hp_metric
在 Metrics
下可见。
我们如何将 train_loss
和 val_loss
添加到 Metrics
部分?这样,我们就可以在 val_loss
中使用 PARALLEL COORDINATES VIEW
而不是 hp_metric
。
显示 hp_metric
而没有 val_loss
的图像:
使用 Pytorch 1.8.1、Pytorch Lightning 1.2.6、TensorBoard 2.4.1
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。