| 97 | print('-------------- End ----------------') |
| 98 | |
| 99 | def _save(self, args): |
| 100 | expr_dir = os.path.join(self._opt.checkpoints_dir, self._opt.name) |
| 101 | print(expr_dir) |
| 102 | util.mkdirs(expr_dir) |
| 103 | file_name = os.path.join(expr_dir, 'opt_%s.txt' % ('train' if self.is_train else 'test')) |
| 104 | with open(file_name, 'wt') as opt_file: |
| 105 | opt_file.write('------------ Options -------------\n') |
| 106 | for k, v in sorted(args.items()): |
| 107 | opt_file.write('%s: %s\n' % (str(k), str(v))) |
| 108 | opt_file.write('-------------- End ----------------\n') |