epoch 期间的 Pytorch 闪电保存模型

Question

每次验证循环结束时，我都想保存一个检查点。我设置了val_check_interval成为 0.2所以我在每个时期有 5 个验证循环，但检查点回调仅在时期结束时保存模型。我找不到在每个验证循环后保存模型的简单(或困难)方法。这似乎有点奇怪，因为除了保存检查点之外，我看不到进行验证循环的理由。
这样做的最简单方法是什么？
谢谢，
吉拉德

Answer 1

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