DeepSpeed
estimate_zero2_model_states_mem_needs: fixing memory estiamtion
#5099
Merged
Merged
estimate_zero2_model_states_mem_needs: fixing memory estiamtion #5099
tjruwase
merged
6 commits into
deepspeedai:master
from
nelyahu:fix_memory_estimation
estimate_zero2_model_states_mem_needs: fixing memory estiamtion
13fbabb3
tjruwase
commented
on 2024-02-08
Merge branch 'master' into fix_memory_estimation
7a1e63dc
tjruwase
commented
on 2024-04-13
tjruwase
commented
on 2024-04-13
Update deepspeed/runtime/zero/stage_1_and_2.py
3b7a0533
Update deepspeed/runtime/zero/stage_1_and_2.py
9b1b7e0b
Merge branch 'master' into fix_memory_estimation
cd57043c
tjruwase
approved these changes
on 2024-05-13
Merge branch 'master' into fix_memory_estimation
52f6c39f
Login to write a write a comment.
Login via GitHub