Home
last modified time | relevance | path

Searched defs:CHECKPOINT_DIR (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/checkpoint/
H A Dtest_dtensor_resharding.py20 CHECKPOINT_DIR = "checkpoint" variable
/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/examples/
H A Dfsdp_checkpoint_example.py23 CHECKPOINT_DIR = f"/scratch/{os.environ['LOGNAME']}/checkpoint" variable
H A Dasync_checkpointing_example.py26 CHECKPOINT_DIR = f"~/{os.environ.get('LOGNAME', '')}/checkpoint" variable
H A Dstateful_example.py23 CHECKPOINT_DIR = f"~/{os.environ['LOGNAME']}/checkpoint" variable