diff --git a/Training/training.py b/Training/training.py index a3c045c..f87301e 100644 --- a/Training/training.py +++ b/Training/training.py @@ -84,7 +84,7 @@ def get_model_memory_usage(batch_size, model): help='overwrite output folder', default=False) parser.add_argument('-p', '--parametric', action='store_true', dest='parametric', - help='train a parametric model', default=True) + help='train a parametric model', default=False) parser.add_argument('--opt', action='store_true',default=False, dest='opt', help='optimize training through back stepping on class loss')