diff --git a/downloads-generation/models_class1_unselected/GENERATE.sh b/downloads-generation/models_class1_unselected/GENERATE.sh index 124715fae80a1c3a60c85b00713d470058a3625b..f04e23ddf9d603a6a255a86b1b9d0f6bf132e241 100755 --- a/downloads-generation/models_class1_unselected/GENERATE.sh +++ b/downloads-generation/models_class1_unselected/GENERATE.sh @@ -34,13 +34,16 @@ python $SCRIPT_DIR/generate_hyperparameters.py > hyperparameters.yaml GPUS=$(nvidia-smi -L 2> /dev/null | wc -l) || GPUS=0 echo "Detected GPUS: $GPUS" +PROCESSORS=$(getconf _NPROCESSORS_ONLN) +echo "Detected processors: $PROCESSORS" + 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 $GPUS --max-workers-per-gpu 2 + --num-jobs $(expr $PROCESSORS \* 2) --gpus $GPUS --max-workers-per-gpu 2 cp $SCRIPT_ABSOLUTE_PATH . bzip2 LOG.txt