MCPcopy Create free account
hub / github.com/apple/axlearn / parse_step_from_dir

Function parse_step_from_dir

axlearn/common/checkpointer.py:117–119  ·  view source on GitHub ↗
(step_dir: str)

Source from the content-addressed store, hash-verified

115
116
117def parse_step_from_dir(step_dir: str) -> int:
118 # TODO(markblee): use regex.
119 return int(step_dir[-STEP_NUM_DIGITS:])
120
121
122def build_step_dir(base_dir: str, *, step: int) -> str:

Callers 5

mainFunction · 0.90
__call__Method · 0.90
checkpoint_stepsMethod · 0.85
restoreMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected