Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
model-checkpointing.rst 350 Bytes
Newer Older
Jeff Rasley's avatar
ZeRO-2 (#217)  
Jeff Rasley committed
1
2
3
4
5
6
7
8
9
10
11
12 
Model Checkpointing
===================

DeepSpeed provides routines for checkpointing model state during training.

Loading Training Checkpoints
----------------------------
.. autofunction:: deepspeed.DeepSpeedLight.load_checkpoint

Saving Training Checkpoints
---------------------------
.. autofunction:: deepspeed.DeepSpeedLight.save_checkpoint