Update main.py

Whoo-jl · Oct 4, 2020 · f5b887b · f5b887b
1 parent eba7ec1
commit f5b887b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -27,7 +27,7 @@ def main(config, args):
     parser.add_argument('--batch_size', type=int, default=None, help='')
     parser.add_argument('--resume', type=str, default=None, help='')
     parser.add_argument('--vis_dir', type=str, default=None, help='')
-    parser.add_argument('--rec', default='aster', choices=['aster, moran, crnn'])
+    parser.add_argument('--rec', default='aster', choices=['aster', 'moran', 'crnn'])
     parser.add_argument('--STN', action='store_true', default=False, help='')
     parser.add_argument('--syn', action='store_true', default=False, help='use synthetic LR')
     parser.add_argument('--mixed', action='store_true', default=False, help='mix synthetic with real LR')