Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Menu
Open sidebar
nlp
ahisto-modules
Named Entity Recognition Experiments
Commits
a3022d22
Commit
a3022d22
authored
Aug 15, 2022
by
Vít Novotný
Browse files
Start taking GPUs from the tail of IDs
parent
eb609384
Pipeline
#146496
passed with stage
in 9 minutes
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
scripts/03_train_ner_models.sh
View file @
a3022d22
...
...
@@ -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
\
--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
"
'
'
:::: scripts/03_train_ner_models.tasks
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment