JayceAnova commited on
Commit
ff8573a
·
verified ·
1 Parent(s): 095abb0

Update instruments_eval.sh

Browse files
Files changed (1) hide show
  1. instruments_eval.sh +6 -5
instruments_eval.sh CHANGED
@@ -1,9 +1,10 @@
 
1
  DATASET=Instruments
2
- BASE=/datain/v-yinju/llama-7b
3
- DATA_PATH=/datain/v-yinju/rqvae-zzx/data
4
- CKPT_PATH=/datain/v-yinju/RQVAE_Bench/llama
5
  RESULTS_FILE=$CKPT_PATH/result.json
6
- INDEX=/datain/v-yinju/RQVAE_Bench/rqvae/Nov-27-2024_23-08-08/indices.json
7
 
8
  torchrun --nproc_per_node=8 test_ddp.py \
9
  --base_model $BASE \
@@ -12,6 +13,6 @@ torchrun --nproc_per_node=8 test_ddp.py \
12
  --data_path $DATA_PATH \
13
  --results_file $RESULTS_FILE \
14
  --test_batch_size 1 \
15
- --num_beams 10 \
16
  --test_prompt_ids all \
17
  --index_file $INDEX
 
1
+ ROOT=/home/jovyan/workspace
2
  DATASET=Instruments
3
+ BASE=$ROOT/Llama-2-7b-hf
4
+ DATA_PATH=$ROOT
5
+ CKPT_PATH=./Ins/Llama-2-7b
6
  RESULTS_FILE=$CKPT_PATH/result.json
7
+ INDEX=$ROOT/$DATASET/Instruments.index.json
8
 
9
  torchrun --nproc_per_node=8 test_ddp.py \
10
  --base_model $BASE \
 
13
  --data_path $DATA_PATH \
14
  --results_file $RESULTS_FILE \
15
  --test_batch_size 1 \
16
+ --num_beams 20 \
17
  --test_prompt_ids all \
18
  --index_file $INDEX