Disables shared-memory workers by default, see #126

mapbox · Oct 9, 2018 · b7f6ebf · b7f6ebf
1 parent dcea15d
commit b7f6ebf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/robosat/tools/predict.py b/robosat/tools/predict.py
@@ -31,7 +31,7 @@ def add_parser(subparser):
     parser.add_argument("--checkpoint", type=str, required=True, help="model checkpoint to load")
     parser.add_argument("--overlap", type=int, default=32, help="tile pixel overlap to predict on")
     parser.add_argument("--tile_size", type=int, required=True, help="tile size for slippy map tiles")
-    parser.add_argument("--workers", type=int, default=1, help="number of workers pre-processing images")
+    parser.add_argument("--workers", type=int, default=0, help="number of workers pre-processing images")
     parser.add_argument("tiles", type=str, help="directory to read slippy map image tiles from")
     parser.add_argument("probs", type=str, help="directory to save slippy map probability masks to")
     parser.add_argument("--model", type=str, required=True, help="path to model configuration file")

diff --git a/robosat/tools/train.py b/robosat/tools/train.py
@@ -48,7 +48,7 @@ def add_parser(subparser):
     parser.add_argument("--dataset", type=str, required=True, help="path to dataset configuration file")
     parser.add_argument("--checkpoint", type=str, required=False, help="path to a model checkpoint (to retrain)")
     parser.add_argument("--resume", type=bool, default=False, help="resume training or fine-tuning (if checkpoint)")
-    parser.add_argument("--workers", type=int, default=1, help="number of workers pre-processing images")
+    parser.add_argument("--workers", type=int, default=0, help="number of workers pre-processing images")
 
     parser.set_defaults(func=main)