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) model.save_weights(trainer.outputs_config['weights'])