Commit a3022d22 authored by Vít Starý Novotný's avatar Vít Starý Novotný
Browse files

Start taking GPUs from the tail of IDs

parent eb609384

scripts/03_train_ner_models.sh

+1 −1
Original line number Original line Diff line number Diff line
@@ -16,6 +16,6 @@ parallel --halt=soon,fail=100% --jobs=4 --bar --delay 60 --resume-failed \ 
         --joblog scripts/03_train_ner_models_"${SCHEDULE_NAME}".joblog \

         --joblog scripts/03_train_ner_models_"${SCHEDULE_NAME}".joblog \

         --colsep ' +' \

         --colsep ' +' \

         -- \

         -- \

           'GPU_ID=$(nvidia-smi | grep -F -B 1 -- "0MiB / 15360MiB" | head -n 1 | awk "{ print \$2 }"); '\

           'GPU_ID=$(nvidia-smi | grep -F -B 1 -- "0MiB / 15360MiB" | tail -n 2 | head -n 1 | awk "{ print \$2 }"); '\

           'docker run --rm -u "$(id -u):$(id -g)" --hostname "'"$HOSTNAME"'" --runtime=nvidia -e CUDA_DEVICE_ORDER=PCI_BUS_ID -e NVIDIA_VISIBLE_DEVICES="$GPU_ID" -e TOKENIZERS_PARALLELISM=false -e COMET_API_KEY -v "$PWD"/..:/workdir:rw -w /workdir/"${PWD##*/}" -v "'"$ROOT_PATH"'":"'"$ROOT_PATH"'":rw -v "'"$ANNOTATION_PATH"'":"'"$ANNOTATION_PATH"'":ro -v "'"$OCR_EVAL_PATH"'":"'"$OCR_EVAL_PATH"'":ro "'"$IMAGE_NAME"'" nice -n 19 python scripts/03_train_ner_models.py {1} {2} {3} '"$SCHEDULE_NAME"'

           'docker run --rm -u "$(id -u):$(id -g)" --hostname "'"$HOSTNAME"'" --runtime=nvidia -e CUDA_DEVICE_ORDER=PCI_BUS_ID -e NVIDIA_VISIBLE_DEVICES="$GPU_ID" -e TOKENIZERS_PARALLELISM=false -e COMET_API_KEY -v "$PWD"/..:/workdir:rw -w /workdir/"${PWD##*/}" -v "'"$ROOT_PATH"'":"'"$ROOT_PATH"'":rw -v "'"$ANNOTATION_PATH"'":"'"$ANNOTATION_PATH"'":ro -v "'"$OCR_EVAL_PATH"'":"'"$OCR_EVAL_PATH"'":ro "'"$IMAGE_NAME"'" nice -n 19 python scripts/03_train_ner_models.py {1} {2} {3} '"$SCHEDULE_NAME"'

         ' :::: scripts/03_train_ner_models.tasks
 
         ' :::: scripts/03_train_ner_models.tasks