Skip to content
New issue

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

Sign up for GitHub

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

Already on GitHub? Sign in to your account

Jump to bottom

estimate_zero2_model_states_mem_needs: fixing memory estiamtion #5099

Merged
merged 6 commits into from
Jun 4, 2024
Merged

estimate_zero2_model_states_mem_needs: fixing memory estiamtion #5099

merged 6 commits into from
Jun 4, 2024

Commits on Feb 8, 2024

  1. estimate_zero2_model_states_mem_needs: fixing memory estiamtion 

    was considering 4 bytes per model param, and 4 bytes per gradient.
fixed it to 2 bytes - under the assumption of FP16/BF16
    @nelyahu
    nelyahu committed Feb 8, 2024
    Configuration menu
    Copy the full SHA
    13fbabb View commit details
    Browse the repository at this point in the history

Commits on Feb 9, 2024

  1. Merge branch 'master' into fix_memory_estimation

    @tjruwase
    tjruwase authored Feb 9, 2024
    Configuration menu
    Copy the full SHA
    7a1e63d View commit details
    Browse the repository at this point in the history

Commits on Apr 22, 2024

  1. Update deepspeed/runtime/zero/stage_1_and_2.py 

    Applying code review suggestion to have more close estimation of memory consumption

Co-authored-by: Olatunji Ruwase <[email protected]>
    @nelyahu @tjruwase
    nelyahu and tjruwase authored Apr 22, 2024
    Configuration menu
    Copy the full SHA
    3b7a053 View commit details
    Browse the repository at this point in the history

  2. Update deepspeed/runtime/zero/stage_1_and_2.py 

    update comment

Co-authored-by: Olatunji Ruwase <[email protected]>
    @nelyahu @tjruwase
    nelyahu and tjruwase authored Apr 22, 2024
    Configuration menu
    Copy the full SHA
    9b1b7e0 View commit details
    Browse the repository at this point in the history

Commits on May 13, 2024

  1. Merge branch 'master' into fix_memory_estimation

    @tjruwase
    tjruwase authored May 13, 2024
    Configuration menu
    Copy the full SHA
    cd57043 View commit details
    Browse the repository at this point in the history

Commits on May 28, 2024

  1. Merge branch 'master' into fix_memory_estimation

    @tjruwase
    tjruwase authored May 28, 2024
    Configuration menu
    Copy the full SHA
    52f6c39 View commit details
    Browse the repository at this point in the history