MCPcopy
hub / github.com/deepspeedai/DeepSpeed / get_optim_files

Function get_optim_files

deepspeed/utils/zero_to_fp32.py:94–95  ·  view source on GitHub ↗
(checkpoint_dir)

Source from the content-addressed store, hash-verified

92
93
94def get_optim_files(checkpoint_dir):
95 return get_checkpoint_files(checkpoint_dir, "*_optim_states.pt")
96
97
98def get_model_state_files(checkpoint_dir):

Callers 2

test_get_optim_filesFunction · 0.90
_get_fp32_state_dict_from_zero_checkpointFunction · 0.85

Calls 1

get_checkpoint_filesFunction · 0.85

Tested by 1

test_get_optim_filesFunction · 0.72

Used in the wild real call sites across dependent graphs

searching dependent graphs…