zzsfornlp / zmsp Goto Github PK

#!/usr/bin/env bash
# =====
# base options
base_opt="conf_output:_conf partype:graph init_from_pretrain:1 lrate.val:0.001 drop_embed:0.33 dropmd_embed:0.33 singleton_thr:2 fix_drop:1 max_epochs:300"
global_unproj_m="dec_algorithm:unproj output_normalizing:global"
# possible losses
prob_loss="loss_function:prob margin.init_val:0."

# =====
# specify paths
SRC_DIR=../v1/
method=${global_unproj_m}
loss=${prob_loss}
# further specify `train/dev/test/embed` paths
DATA_TRAIN="data/2.7UD_English-EWT/en_ewt-ud-train.conllu"
DATA_DEV="data/2.7UD_English-EWT/en_ewt-ud-dev.conllu"
DATA_TEST="data/2.7UD_English-EWT/en_ewt-ud-test.conllu"
DATA_EMBED="data/wiki.en.vec"
# =====
# running
# on GPU 0
CUDA_VISIBLE_DEVICES=0 PYTHONPATH=${SRC_DIR} python3 v1/tasks/cmd.py zdpar.main.train train:${DATA_TRAIN} dev:${DATA_DEV} test:${DATA_TEST} pretrain_file:${DATA_EMBED} device:0 ${base_opt} ${method} ${loss}