forked from AudioLLMs/AudioBench
-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_si.sh
31 lines (28 loc) · 875 Bytes
/
run_si.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
export AZURE_OPENAI_KEY=""
export OPENAI_API_KEY=""
# = = = = = = = = = = = = = = =
MODEL_PATH_OR_ID="/home/root/BachVD/model_zoo/llama3.1-s-instruct-2024-08-19-epoch-3/"
BATCH_SIZE=1
OVERWRITE=False
NUMBER_OF_SAMPLES=-1
# = = = = = = = = = = = = = = =
export CUDA_VISIBLE_DEVICES=0
# SI
DATASET=openhermes_audio_test
METRICS=gpt4_judge
python src/main_evaluate.py \
--dataset_name $DATASET \
--model_path_or_id $MODEL_PATH_OR_ID \
--batch_size $BATCH_SIZE \
--overwrite $OVERWRITE \
--metrics $METRICS \
--number_of_samples $NUMBER_OF_SAMPLES
DATASET=alpaca_audio_test
METRICS=gpt4_judge
python src/main_evaluate.py \
--dataset_name $DATASET \
--model_path_or_id $MODEL_PATH_OR_ID \
--batch_size $BATCH_SIZE \
--overwrite $OVERWRITE \
--metrics $METRICS \
--number_of_samples $NUMBER_OF_SAMPLES