draugr.torch_utilities.persistence.checkpoint.save_checkpoint

draugr.torch_utilities.persistence.checkpoint.save_checkpoint(PATH: Path, epoch, model, optimiser, loss)[source]
Parameters
  • PATH

  • epoch

  • model

  • optimiser

  • loss