Fix fizyr#840

huynhtnhut97 · Dec 17, 2018 · 5268499 · 5268499
1 parent c8b8ea2
commit 5268499
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keras_retinanet/bin/train.py b/keras_retinanet/bin/train.py
@@ -410,8 +410,8 @@ def csv_list(string):
     parser.add_argument('--weighted-average', help='Compute the mAP using the weighted average of precisions among classes.', action='store_true')
 
     # Fit generator arguments
-    parser.add_argument('--workers', help='Number of multiprocessing workers. To disable multiprocessing, set workers to 0', default=1)
-    parser.add_argument('--max-queue-size', help='Queue length for multiprocessing workers in fit generator.', default=10)
+    parser.add_argument('--workers', help='Number of multiprocessing workers. To disable multiprocessing, set workers to 0', default=1, type=int)
+    parser.add_argument('--max-queue-size', help='Queue length for multiprocessing workers in fit generator.', default=10, type=int)
 
     return check_args(parser.parse_args(args))