core.utils.trainutils.save_model

core.utils.trainutils.save_model(
    pm,
    log_dir,
    model_prefix='model',
    checkpoint_epochs=None,
)

Save model to _latest.pt in log_dir. Additionally, all checkpoint_epochs a permanent checkpoint is saved.

Parameters

Name	Type	Description	Default
pm		model to be saved	required
log_dir		log directory, where model is saved	required
model_prefix		prefix for name of save model	'model'
checkpoint_epochs		number of steps between two consecutive model checkpoints	None