-
Notifications
You must be signed in to change notification settings - Fork 1
/
run_sst.sh
19 lines (16 loc) · 819 Bytes
/
run_sst.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
#!/bin/bash
start_time=$(date +%s)
gpu=0
dataset=sst
DATA_DIR=/your_path/FastClass/CLARET/data/sst/
Sen_DIR=/your_path/FastClass/CLARET/SentenceTransformer/
labels=2
CUDA_VISIBLE_DEVICES=${gpu} python3 SentenceTransformer/Task_data.py --task ${dataset} --path ${DATA_DIR}
CUDA_VISIBLE_DEVICES=${gpu} python3 SentenceTransformer/Task_external_data.py --task ${dataset} --path ${DATA_DIR} --model_path ${Sen_DIR}
CUDA_VISIBLE_DEVICES=${gpu} python3 SentenceTransformer/SubmodularSelection_Mix.py --task ${dataset} --path ${DATA_DIR}
CUDA_VISIBLE_DEVICES=${gpu} python3 -u Select_N_K2.py --num_labels ${labels}\
--train_path ${DATA_DIR}\
--test_corpus ${DATA_DIR}test.txt
end_time=$(date +%s)
cost_time=$[ $end_time-$start_time ]
echo "Total time : $(($cost_time/60))min $(($cost_time%60))s"