graphorge.gnn_base_model.train.training.remove_posterior_optim_state_files

remove_posterior_optim_state_files(model, epoch)[source]

Delete optimizer training epoch state files posterior to given epoch.

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

  • epoch (int) – Training epoch.