parser / udpipe2 /scripts /compute_embeddings.sh
anasampa2's picture
Upload 110 files
0e5da39 verified
raw
history blame
No virus
571 Bytes
#!/bin/sh
[ $# -ge 1 ] || { echo Usage: $0 data_directory embedding_args... >&2; exit 1; }
data="$1"; shift
#for d in $data/*1.0-[a-z]*/; do
for d in $data/*/; do
case $d in
*/cs_*) model=robeczech-base-last4;;
*) model=bert-base-multilingual-uncased-last4;;
esac
for f in $d*.conllu; do
[ $f.npz -nt $f ] && continue
sbatch -p gpu-ms,gpu-troja -G 1 -C "gpu_cc6.1|gpu_cc7.5" --mem=16G -o $f.log run withcuda101 wembedding_service/venv/bin/python wembedding_service/compute_wembeddings.py --format=conllu $f $f.npz --model=$model "$@"
done
done