import argparse
import json

from trainers import trainers


def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--config", required=True)
    return parser.parse_args()


if __name__ == "__main__":
    args = parse_args()
    config = json.loads(open(args.config).read())
    trainer_type = getattr(trainers, config["type"])

    print("Trainer: ", config["type"], trainer_type)
    trainer = trainer_type(config)
    trainer.loop()