Skip to content

Conversation

@samyam
Copy link
Contributor

@samyam samyam commented Mar 10, 2020

… states. Unittest testing saving and loading checkpoint with fused, unfused and zero optimizer. The unitest takes about 165s

@samyam samyam added the enhancement New feature or request label Mar 10, 2020
@samyam samyam requested review from ShadenSmith and jeffra March 10, 2020 20:27
samyam added 2 commits March 10, 2020 20:55
… states. Unittest testing saving and loading checkpoint with fused, unfused and zero optimizer. The unitest takes about 165s
@samyam samyam force-pushed the samyamr/no-load-optim-checkpoint branch from cd3e732 to d2c52cf Compare March 10, 2020 20:56
@samyam samyam linked an issue Mar 10, 2020 that may be closed by this pull request
@jeffra jeffra merged commit 936117b into master Mar 10, 2020
@jeffra jeffra deleted the samyamr/no-load-optim-checkpoint branch March 10, 2020 23:52
kouml pushed a commit to kouml/DeepSpeed that referenced this pull request Apr 3, 2020
deepspeedai#128)

* Enhancement: Ability to load checkpoint without loading the optimizer states. Unittest testing saving and loading checkpoint with fused, unfused and zero optimizer. The unitest takes about 165s
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Load model checkpoint without loading the optimizer states.

3 participants