Skip to content

Commit

Permalink
Modify data path
Browse files Browse the repository at this point in the history
  • Loading branch information
Andreas-Xi authored Nov 8, 2024
1 parent dc5d254 commit f605258
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/drcap_zeroshot_aac/scripts/inference_drcap.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand Down Expand Up @@ -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'
# note: to inference model trained the linear layer only, you could set '++train_config.use_peft=false' and 'train_config.freeze_llm=true'

0 comments on commit f605258

Please sign in to comment.