checkpoint_6.py 187 B

1234
  1. def tag_last_checkpoint(self, last_filename):
  2. save_file = os.path.join(self.save_dir, "last_checkpoint")
  3. with open(save_file, "w") as f:
  4. f.write(last_filename)