Update instruments_eval.sh
Browse files- instruments_eval.sh +6 -5
instruments_eval.sh
CHANGED
|
@@ -1,9 +1,10 @@
|
|
|
|
|
| 1 |
DATASET=Instruments
|
| 2 |
-
BASE=/
|
| 3 |
-
DATA_PATH=
|
| 4 |
-
CKPT_PATH=/
|
| 5 |
RESULTS_FILE=$CKPT_PATH/result.json
|
| 6 |
-
INDEX=/
|
| 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
|
| 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
|