--- a +++ b/inference.py @@ -0,0 +1,20 @@ +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()