From 2aa680d06b5ca6aad698105f5d3a841f97129142 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Fri, 9 Feb 2018 19:38:07 -0500
Subject: [PATCH] update GENERATE
---
downloads-generation/models_class1_unselected/GENERATE.sh | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/downloads-generation/models_class1_unselected/GENERATE.sh b/downloads-generation/models_class1_unselected/GENERATE.sh
index 3a3af72e..124715fa 100755
--- a/downloads-generation/models_class1_unselected/GENERATE.sh
+++ b/downloads-generation/models_class1_unselected/GENERATE.sh
@@ -31,13 +31,16 @@ mkdir models
python $SCRIPT_DIR/generate_hyperparameters.py > hyperparameters.yaml
+GPUS=$(nvidia-smi -L 2> /dev/null | wc -l) || GPUS=0
+echo "Detected GPUS: $GPUS"
+
time mhcflurry-class1-train-allele-specific-models \
--data "$(mhcflurry-downloads path data_curated)/curated_training_data.no_mass_spec.csv.bz2" \
--hyperparameters hyperparameters.yaml \
--out-models-dir models \
--percent-rank-calibration-num-peptides-per-length 0 \
--min-measurements-per-allele 75 \
- --num-jobs 32 --gpus 4 --backend tensorflow-default
+ --num-jobs 32 --gpus $GPUS --max-workers-per-gpu 2
cp $SCRIPT_ABSOLUTE_PATH .
bzip2 LOG.txt
--
GitLab