-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy patheval_acenet_csqa.sh
33 lines (26 loc) · 938 Bytes
/
eval_acenet_csqa.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
32
33
#!/bin/bash
export CUDA_VISIBLE_DEVICES=3
dt=`date '+%Y%m%d_%H%M%S'`
dataset="csqa"
model='roberta-large'
shift $(( $# >0 ? 1 : 0))
shift $(( $# >0 ? 1 : 0))
args=$@
echo "******************************"
echo "dataset: $dataset"
echo "******************************"
save_dir_pref='saved_models'
mkdir -p $save_dir_pref
###### Eval ######
python3 -u acenet_best.py --dataset $dataset \
--train_adj data/${dataset}/graph/train.graph.adj.pk \
--dev_adj data/${dataset}/graph/dev.graph.adj.pk \
--test_adj data/${dataset}/graph/test.graph.adj.pk \
--train_statements data/${dataset}/statement/train.statement.jsonl \
--dev_statements data/${dataset}/statement/dev.statement.jsonl \
--test_statements data/${dataset}/statement/test.statement.jsonl \
--save_model \
--save_dir saved_models \
--mode eval_detail \
--load_model_path saved_models/model.pt.13 \
$args