From 97d4912e405af7120890795b60075d6df3a62b10 Mon Sep 17 00:00:00 2001 From: Sonia Date: Fri, 11 Jun 2021 18:48:13 +0200 Subject: [PATCH] Update utils.py --- utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/utils.py b/utils.py index 8976cd4..c643d39 100644 --- a/utils.py +++ b/utils.py @@ -15,8 +15,10 @@ def parse_args(): parser = argparse.ArgumentParser( description='3D ConvLSTM training', - add_help=True) + add_help=True) + parser.add_argument("--simu_path", '-simu', default='.', type=str, + help="Path pointing to the folder where the MC simulations are kept in .npy format") parser.add_argument("--n_samples", '-n', default=0, type=float, help='Number of training samples') parser.add_argument("--patch_size", '-ps', default=64, type=int,