train stores the path for training configurations.
test stores the path for testing configurations.
gpu stores the gpu id used.
"""
parser = argparse.ArgumentParser()
parser.add_argument('--model', dest='model', type=str, required=True)
parser.add_argument('--train', dest='train', type=str)
parser.add_argument('--test', dest='test', type=str)
args = parser.parse_args()
# Build Model Graph from Config
model_config = configparser.ConfigParser()
model_config.read(args.model)
model = ModelBuilder(model_config)
model.build_graph()
model.compile()
model.summary_txt()
model.print_png()
model.save_graph()
# Train Model
if args.train:
train_config = configparser.ConfigParser()
train_config.read(args.train)
trainer = ModelTrainer(train_config)
trainer.get_hyperparameters()
trainer.get_train_set()
trainer.get_dev_set()
trainer.get_callbacks()
train_history = model.train(trainer.fit_options)
