From 5a1800f2f53ed97d39e981ff1ce0a5665d43424c Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Tue, 30 Jul 2019 12:47:20 -0400
Subject: [PATCH] fix
---
.../models_class1_pan_unselected/GENERATE.sh | 8 +++++++-
1 file changed, 7 insertions(+), 1 deletion(-)
diff --git a/downloads-generation/models_class1_pan_unselected/GENERATE.sh b/downloads-generation/models_class1_pan_unselected/GENERATE.sh
index 1d3cef92..b70d5b8d 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 .
--
GitLab