Update main.py

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)
+    main(arguments)