Pytorch Lightning,在几批后显示验证指标

问题描述 投票:0回答:1

Pytorch Lightning中,我想在日志和终端的每个N步骤(批次)的训练批次和验证数据集上打印lossaccuracy

这个问题有点类似于:计算每 n 批 Pytorch Lightning 的指标/损失

但是最简单的解决方案对我来说不起作用,而另一种解决方案对于这种基本功能来说似乎过于复杂。

pytorch pytorch-lightning
1个回答
0
投票

同时我找到了一个可能的答案,那就是使用:

 val_check_interval=n
选项
Trainer

https://lightning.ai/docs/pytorch/stable/common/trainer.html#val-check-interval

来自文档:

在一个训练周期内检查验证集的频率。可以指定为 float 或 int。

  • 传递 [0.0, 1.0] 范围内的浮点数来检查一小部分之后 训练纪元。

  • 在固定数量的训练批次后传递一个 int 进行检查。

© www.soinside.com 2019 - 2024. All rights reserved.