New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Jump to bottom

Different zero stage the training memory compute #912

Open

Arcmoon-Hu opened this issue Jul 17, 2024 · 0 comments

Arcmoon-Hu commented Jul 17, 2024

If I use zero stage 2 and enable gradient checkpoint then how compute the activation memory ?
There is a ref paper https://arxiv.org/pdf/2205.05198 :

# for free to join this conversation on GitHub. Already have an account? # to comment