|
""" |
|
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() |
|
|