问题描述
我想在每次验证循环结束时保存一个检查点。我将 val_check_interval
设置为 0.2
,因此我在每个时期有 5 个验证循环,但检查点回调仅在时期结束时保存模型。我找不到在每个验证循环后保存模型的简单(或困难)方法。这似乎有点奇怪,因为除了保存检查点之外,我看不到进行验证循环的原因。
最简单的方法是什么?
谢谢,
吉拉德
解决方法
您是否检查过pytorch_lightning.callbacks.model_checkpoint.ModelCheckpoint?
不确定它是否存在于您的版本中,但是将 every_n_val_epochs
设置为 1 应该可以工作。如果是这样,它应该在每个验证循环后保存您的模型检查点。