-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathrun_baseline_pipeline.sh
36 lines (30 loc) · 1.19 KB
/
run_baseline_pipeline.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
34
35
36
#!/bin/bash
lm="deberta-large"
portion="persona"
window="nlg"
# Prepare directories for intermediate results of each subtask
mkdir -p pred/${portion}-${lm}-${window}-pipeline-test
visible=0
CUDA_VISIBLE_DEVICES=${visible} python baseline.py \
--eval_only \
--checkpoint runs/${portion}-${lm}-${window}-head-test/ \
--params_file runs/${portion}-${lm}-${window}-head-test/params-${lm}.json \
--eval_dataset test \
--dataroot data/${portion}/head/${window} \
--output_file pred/${portion}-${lm}-${window}-head-test/predictions.json
CUDA_VISIBLE_DEVICES=${visible} python data_preprocessing_pipeline_test.py \
--model ${lm} \
--window ${window} \
--portion ${portion}
CUDA_VISIBLE_DEVICES=${visible} python baseline.py \
--eval_only \
--checkpoint runs/${portion}-${lm}-${window}-fact_cut-test/ \
--params_file runs/${portion}-${lm}-${window}-fact_cut-test/params-${lm}.json \
--eval_dataset test \
--dataroot data/${portion}/fact_pipe/${window} \
--output_file pred/${portion}-${lm}-${window}-pipeline-test/predictions.json
CUDA_VISIBLE_DEVICES=${visible} python evaluate_pipeline.py \
--model ${lm} \
--window ${window} \
--portion ${portion} \
--linking fact