| """ | |
| This file runs the main training/val loop | |
| """ | |
| import os | |
| import json | |
| import sys | |
| import pprint | |
| sys.path.append(".") | |
| sys.path.append("..") | |
| from mapper.options.train_options import TrainOptions | |
| from mapper.training.coach import Coach | |
| def main(opts): | |
| if os.path.exists(opts.exp_dir): | |
| raise Exception('Oops... {} already exists'.format(opts.exp_dir)) | |
| os.makedirs(opts.exp_dir, exist_ok=True) | |
| 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() | |
| if __name__ == '__main__': | |
| opts = TrainOptions().parse() | |
| main(opts) | |