Skip to content
Snippets Groups Projects
Commit 1acaf1c6 authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

update generate

parent 2aa680d0
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment