| """ | |
| This file runs the main training/val loop | |
| """ | |
| import os | |
| os.environ["CUDA_VISIBLE_DEVICES"] = "2" | |
| import json | |
| import sys | |
| import pprint | |
| sys.path.append(".") | |
| sys.path.append("..") | |
| from options.train_options import TrainOptions | |
| from training.coach import Coach | |
| # from training.big_angle_coach import BA_Coach | |
| def main(): | |
| opts = TrainOptions().parse() | |
| # if os.path.exists(opts.exp_dir): | |
| #raise Exception('Oops... {} already exists'.format(opts.exp_dir)) | |
| # os.makedirs(opts.exp_dir) | |
| if not os.path.exists(opts.exp_dir): | |
| os.mkdir(opts.exp_dir) | |
| opts_dict = vars(opts) | |
| pprint.pprint(opts_dict) | |
| with open(os.path.join(opts.exp_dir, 'opt.json'), 'w') as f: | |
| json.dump(opts_dict, f, indent=4, sort_keys=True) | |
| coach = Coach(opts) | |
| # coach.train() | |
| coach.train_encoder() | |
| if __name__ == '__main__': | |
| main() | |