cerebras.modelzoo.common.pytorch_utils.get_checkpoints#

cerebras.modelzoo.common.pytorch_utils.get_checkpoints(model_dir)[source]#

Gather checkpoints in a model directory.