DATASET=Instruments | |
BASE=/datain/v-yinju/llama-7b | |
# DATA_PATH=/datain/v-yinju/rqvae-zzx/data | |
INTER=/datain/v-yinju/LLMBased_Multimodal_RS/Data/Musical_Instruments/Musical_Instruments.inters.numerical.json | |
INDEX=/datain/v-yinju/LLMBased_Multimodal_RS/Data/Musical_Instruments/Rqvae/Text/indices.json | |
CKPT_PATH=/datain/v-yinju/LLMBased_Multimodal_RS/Model_LLaMA/Text | |
RESULTS_FILE=$CKPT_PATH/result.json | |
torchrun --nproc_per_node=8 test_ddp.py \ | |
--base_model $BASE \ | |
--ckpt_path $CKPT_PATH \ | |
--inter_path $INTER \ | |
--dataset $DATASET \ | |
--results_file $RESULTS_FILE \ | |
--test_batch_size 1 \ | |
--num_beams 10 \ | |
--test_prompt_ids all \ | |
--index_file $INDEX |