From 495f9719cb40dd581afdbca1de7894dbe484360d Mon Sep 17 00:00:00 2001 From: Killian Sheriff Date: Mon, 9 Mar 2020 11:30:11 -0400 Subject: [PATCH] Fix missing OutputPaths error --- predict.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/predict.py b/predict.py index 46ff358..baefd51 100644 --- a/predict.py +++ b/predict.py @@ -8,7 +8,7 @@ from config import main_config from models import gmcnn_gan -from utils import training_utils +from utils import training_utils, constants log = training_utils.get_logger() @@ -49,6 +49,9 @@ def main(): parser.add_argument('--save_to', default='predicted.jpg', help='The save path of predicted image') + parser.add_argument('--exp_name', + requiered=True, + help='name of the experiment') args = parser.parse_args() @@ -59,7 +62,7 @@ def main(): img_width=config.training.img_width, num_channels=config.training.num_channels, warm_up_generator=False, - config=config) + config=config,output_paths=constants.OutputPaths(args.exp_name)) log.info('Loading GMCNN model...') gmcnn_model.load() log.info('GMCNN model successfully loaded.')