hookeai.model_architectures.procedures.model_state_files.remove_posterior_state_files

remove_posterior_state_files(model, epoch)[source]

Delete model training epoch state files posterior to given epoch.

Parameters:
  • model (torch.nn.Module) – Model.

  • epoch (int) – Training epoch.