[PyTorch Lightning]:断点续训
要在 PyTorch Lightning 中从断点继续训练,可以使用以下步骤:
1. 保存断点
在训练过程中使用 ModelCheckpoint
回调来保存模型的状态。可以在 Trainer
中设置 checkpoint_callback
参数来使用该回调。
from pytorch_lightning.callbacks import ModelCheckpoint
checkpoint_callback = ModelCheckpoint(
filepath='model-{epoch:02d}-{val_loss:.2f}',
save_top_k=3,
verbose=