forked from Shuailong/BlamePipeline
-
Notifications
You must be signed in to change notification settings - Fork 0
/
coda_run.sh
31 lines (30 loc) · 852 Bytes
/
coda_run.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
CMD="python train.py \
--model-dir . \
--data-dir . \
--embed-dir embed-dir \
--train-file train.json \
--dev-file dev.json \
--test-file test.json \
--pretrain-file elmo \
--fix-embeddings True \
--unk-entity False \
--xavier-init True \
--early-stopping 5 \
--batch-size 1 \
--test-batch-size 1 \
--display-iter 1250 \
--stats-file True"
echo $CMD
cl run \
train.py:script/blameextract/train.py \
train.json:directed-dataset/samples-directed-train.json \
dev.json:directed-dataset/samples-directed-dev.json \
test.json:directed-dataset/samples-directed-test.json \
embed-dir:elmo-weights \
"$CMD" \
--request-docker-image shuailongliang/blamepipeline:0.3.1 \
--request-gpus 1 \
--request-time 2d \
--request-disk 2g \
--request-memory 32g \
-n run-elmo