From f60525862a5327e727a076a26f9139613cf74281 Mon Sep 17 00:00:00 2001 From: Xiquan Li <123717620+Andreas-Xi@users.noreply.github.com> Date: Fri, 8 Nov 2024 23:22:54 +0100 Subject: [PATCH] Modify data path --- examples/drcap_zeroshot_aac/scripts/inference_drcap.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/drcap_zeroshot_aac/scripts/inference_drcap.sh b/examples/drcap_zeroshot_aac/scripts/inference_drcap.sh index 76e99130..6815b7ac 100644 --- a/examples/drcap_zeroshot_aac/scripts/inference_drcap.sh +++ b/examples/drcap_zeroshot_aac/scripts/inference_drcap.sh @@ -18,7 +18,7 @@ pd_text_support=$audio_encoder_dir/support_embeddings/audiocaps_text_support.pt encoder_projector_ds_rate=1 num_beams=4 -inference_data_path=examples/drcap_zeroshot_aac/data/audiocaps_test.jsonl +inference_data_path=examples/drcap_zeroshot_aac/data_examples/audiocaps_test.jsonl decode_log=$output_dir/decode_log_test_clean_beam${num_beams}_repetition_penalty1 @@ -58,4 +58,4 @@ python $code_dir/inference_drcap_batch.py \ ++peft_ckpt=$output_dir \ ++train_config.use_peft=true \ -# note: to inference model trained the linear layer only, you could set '++train_config.use_peft=false' and 'train_config.freeze_llm=true' \ No newline at end of file +# note: to inference model trained the linear layer only, you could set '++train_config.use_peft=false' and 'train_config.freeze_llm=true'