From ab9a1eeaff2bd4850695deb30b0a13597327a479 Mon Sep 17 00:00:00 2001 From: Rodney LaLonde Date: Tue, 8 May 2018 01:34:53 -0400 Subject: [PATCH] Update main.py --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 9cc2e78..f175315 100644 --- a/main.py +++ b/main.py @@ -100,7 +100,7 @@ def main(args): help='/path/to/trained_model.hdf5 from root. Set to "" for none.') parser.add_argument('--split_num', type=int, default=0, help='Which training split to train/test on.') - parser.add_argument('--net', type=str.lower, default='segscapsr3', + parser.add_argument('--net', type=str.lower, default='segcapsr3', choices=['segcapsr3', 'segcapsr1', 'segcapsbasic', 'unet', 'tiramisu'], help='Choose your network.') parser.add_argument('--train', type=int, default=1, choices=[0,1], @@ -172,4 +172,4 @@ def main(args): 'for multi-GPU training. For model parallelism instead of ' \ 'data parallelism, modifications must be made to the code.' - main(arguments) \ No newline at end of file + main(arguments)