-
Notifications
You must be signed in to change notification settings - Fork 4
/
Ex2_three_class_run.sh
executable file
·15 lines (11 loc) · 1.08 KB
/
Ex2_three_class_run.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#!/bin/bash
# Train classifier first
echo "Training classifiers for Ex2b: Three-class Classification."
# CUDA_VISIBLE_DEVICES=0 python benchmark.py --dataset All --Mixcase_filename "2llama_complete.json" --train_with_mixcase --log_name "Ex2b_three_class_train" --ckpt_dir "./ckpt/Ex2b_three_class" --seed 0 --mixcase_threshold 1 --train_threshold 10000 --finetune --three_classes
filenames=("2llama_complete.json" "2llama_rewrite.json" "2llama_polish_token.json" "2llama_polish_sentence.json" "gpt4_complete.json" "gpt4_rewrite.json" "gpt4_polish_token.json" "gpt4_polish_sentence.json" "gpt4_humanize.json" "MGT_polish_sentence.json" "MGT_polish_token.json" "2llama_humanize.json") #
for filename in "${filenames[@]}"
do
echo "Running benchmark for $filename."
CUDA_VISIBLE_DEVICES=0 python benchmark.py --dataset All --Mixcase_filename "${filename}" --log_name "new_Ex2b_three_class_${filename}l" --ckpt_dir "./ckpt/Ex2b_three_class" --seed 0 --mixcase_threshold 1 --train_threshold 10000 --train_with_mixcase --finetune --three_classes --test_only
done
echo "All benchmarks completed."