diff --git a/robosat/tools/predict.py b/robosat/tools/predict.py index 05416913..36436b52 100644 --- 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 index 61a4a678..85245fd8 100644 --- 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)