| 12345678910111213141516 |
- def save_checkpoint(state, is_best, file_path, file_name='checkpoint.pth.tar'):
- """
- Saves the current state of the model. Does a copy of the file
- in case the model performed better than previously.
- Parameters:
- state (dict): Includes optimizer and model state dictionaries.
- is_best (bool): True if model is best performing model.
- file_path (str): Path to save the file.
- file_name (str): File name with extension (default: checkpoint.pth.tar).
- """
- save_path = os.path.join(file_path, file_name)
- torch.save(state, save_path)
- if is_best:
- shutil.copyfile(save_path, os.path.join(file_path, 'model_best.pth.tar'))
|
