import argparse

import json

from runners import runners

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

    runner_type = getattr(runners, config["type"])

    print("Trainer: ", config["type"], runner_type)

    runner = runner_type(config)

    runner.inference()