(state, is_best, save_dir, model_name="")
| 35 | |
| 36 | |
| 37 | def save_checkpoint(state, is_best, save_dir, model_name=""): |
| 38 | if not os.path.exists(save_dir): |
| 39 | os.makedirs(save_dir) |
| 40 | filename = os.path.join(save_dir, model_name + "_ckpt.pth.tar") |
| 41 | torch.save(state, filename) |
| 42 | if is_best: |
| 43 | best_filename = os.path.join(save_dir, "best_ckpt.pth.tar") |
| 44 | shutil.copyfile(filename, best_filename) |