在Pytorch Lightning中,我想在日志和终端的每个N步骤(批次)的训练批次和验证数据集上打印loss和accuracy。
这个问题有点类似于:计算每 n 批 Pytorch Lightning 的指标/损失
但是最简单的解决方案对我来说不起作用,而另一种解决方案对于这种基本功能来说似乎过于复杂。
同时我找到了一个可能的答案,那就是使用:
val_check_interval=n
选项Trainer
。
https://lightning.ai/docs/pytorch/stable/common/trainer.html#val-check-interval
来自文档:
在一个训练周期内检查验证集的频率。可以指定为 float 或 int。
传递 [0.0, 1.0] 范围内的浮点数来检查一小部分之后 训练纪元。
在固定数量的训练批次后传递一个 int 进行检查。