epoch 期间的 Pytorch 闪电保存模型

如何解决epoch 期间的 Pytorch 闪电保存模型

我想在每次验证循环结束时保存一个检查点。我将 val_check_interval 设置为 0.2，因此我在每个时期有 5 个验证循环，但检查点回调仅在时期结束时保存模型。我找不到在每个验证循环后保存模型的简单（或困难）方法。这似乎有点奇怪，因为除了保存检查点之外，我看不到进行验证循环的原因。

最简单的方法是什么？

谢谢，

吉拉德

解决方法

您是否检查过pytorch_lightning.callbacks.model_checkpoint.ModelCheckpoint？ 不确定它是否存在于您的版本中，但是将 every_n_val_epochs 设置为 1 应该可以工作。如果是这样，它应该在每个验证循环后保存您的模型检查点。