[pytorch_lightning]: resume training

阅读量：

[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=True,
    monitor='val_loss',
    mode='min'
    )
    
    trainer = Trainer(
    checkpoint_callback=checkpoint_callback,
    ...
    )

全部评论 (0)

还没有任何评论哟~