| """ |
| This file runs the main training/val loop |
| """ |
| import os |
| import json |
| import sys |
| import pprint |
|
|
| sys.path.append(".") |
| sys.path.append("..") |
|
|
| from options.train_options import TrainOptions |
| from training.coach_aging import Coach |
|
|
|
|
| def main(): |
| opts = TrainOptions().parse() |
| 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__': |
| main() |
|
|
