cerebras.modelzoo.common.pytorch_utils.setup_artifact_dir

cerebras.modelzoo.common.pytorch_utils.setup_artifact_dir(model_dir: str, mode: str)

Create a unique subdirectory for this run by generating a time stamp so that parallel runs using the same model_dir don’t overwrite common files.

previous

cerebras.modelzoo.common.pytorch_utils.load_from_checkpoint_file

next

cerebras.modelzoo.common.pytorch_utils.setup_logging