Heimdall.trainer.HeimdallTrainer.load_checkpoint

Heimdall.trainer.HeimdallTrainer.load_checkpoint#

HeimdallTrainer.load_checkpoint(specific_milestone=None)[source]#

Load a checkpoint based on milestone.txt or a specific milestone number.