MCPcopy Create free account
hub / github.com/zai-org/CodeGeeX / _init_autoresume

Function _init_autoresume

codegeex/megatron/initialize.py:282–288  ·  view source on GitHub ↗

Set autoresume start time.

()

Source from the content-addressed store, hash-verified

280
281
282def _init_autoresume():
283 """Set autoresume start time."""
284 autoresume = get_adlr_autoresume()
285 if autoresume:
286 torch.distributed.barrier()
287 autoresume.init()
288 torch.distributed.barrier()
289
290
291def _set_random_seed(seed_):

Callers 1

initialize_megatronFunction · 0.85

Calls 1

get_adlr_autoresumeFunction · 0.90

Tested by

no test coverage detected