Skip to content

GitLab

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

DeepSpeed provides routines for checkpointing model state during training.

Loading Training Checkpoints
----------------------------
Shaden Smith's avatar
readthedocs upgrade (#402)  
Shaden Smith committed
8 
.. autofunction:: deepspeed.DeepSpeedEngine.load_checkpoint
Jeff Rasley's avatar
ZeRO-2 (#217)  
Jeff Rasley committed
9
10
11 

Saving Training Checkpoints
---------------------------
Shaden Smith's avatar
readthedocs upgrade (#402)  
Shaden Smith committed
12 
.. autofunction:: deepspeed.DeepSpeedEngine.save_checkpoint