diff --git a/downloads-generation/models_class1_pan_unselected/GENERATE.sh b/downloads-generation/models_class1_pan_unselected/GENERATE.sh index 1d3cef92399c845ea57e542ac4e88049131691ed..b70d5b8d3d1a82aa911f68b504cce1d4b5ba6d2d 100755 --- a/downloads-generation/models_class1_pan_unselected/GENERATE.sh +++ b/downloads-generation/models_class1_pan_unselected/GENERATE.sh @@ -34,6 +34,12 @@ echo "Detected GPUS: $GPUS" PROCESSORS=$(getconf _NPROCESSORS_ONLN) echo "Detected processors: $PROCESSORS" +NUM_JOBS=$GPUS +if [ "$NUM_JOBS" -eq "0" ]; then + NUM_JOBS=1 +fi +echo "Num jobs: $NUM_JOBS" + export PYTHONUNBUFFERED=1 for kind in with_mass_spec no_mass_spec @@ -48,7 +54,7 @@ do --out-models-dir models.${kind} \ --worker-log-dir "$SCRATCH_DIR/$DOWNLOAD_NAME" \ --verbosity 0 \ - --num-jobs $GPUS --max-tasks-per-worker 1 --gpus $GPUS --max-workers-per-gpu 1 + --num-jobs $NUM_JOBS --max-tasks-per-worker 1 --gpus $GPUS --max-workers-per-gpu 1 done cp $SCRIPT_ABSOLUTE_PATH .