Pytorch Lightning如何在训练过程中保存模型

Question

Pytorch Lightning如何在训练过程中保存模型

4

每次验证循环结束时，我想保存一个检查点。我将val_check_interval设置为0.2，所以在每个时期内有5个验证循环，但是检查点回调仅在时期结束时保存模型。我找不到一种简单（或难）的方法来在每个验证循环后保存模型。这似乎有些奇怪，因为我看不出除了保存检查点之外还有什么理由去进行验证循环。

最简单的方法是什么？

谢谢，

Gilad

- Gilad

2个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- kkirtac · Answer 1

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

- Vaibhav Joshi · Answer 2

来自闪电文档： save_on_train_epoch_end（Optional[bool]）- 是否在训练周期结束时运行检查点。如果为False，则在验证结束时运行检查。

在训练器的回调中使用save_on_train_epoch_end = False标志应该可以解决这个问题。