Heimdall.trainer.HeimdallTrainer.save_checkpoint

Heimdall.trainer.HeimdallTrainer.save_checkpoint#

HeimdallTrainer.save_checkpoint(epoch)[source]#

Save model checkpoint at the given epoch.