File size: 807 Bytes
6551065 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
#!/bin/bash
dataset="iu_xray"
annotation="data/iu_xray/annotation.json"
base_dir="./data/iu_xray/images"
delta_file="/apdcephfs/share_733425/vinnylywang/zhanyuwang/Code/R2GenGPT/save/iu_xray/v1_delta/checkpoints/checkpoint_epoch13_step1806_bleu0.161532_cider0.530213.pth"
version="v1_delta"
savepath="./save/$dataset/$version"
python -u train.py \
--test \
--dataset ${dataset} \
--annotation ${annotation} \
--base_dir ${base_dir} \
--delta_file ${delta_file} \
--test_batch_size 16 \
--freeze_vm True \
--vis_use_lora True \
--savedmodel_path ${savepath} \
--max_length 60 \
--min_new_tokens 40 \
--max_new_tokens 100 \
--repetition_penalty 2.0 \
--length_penalty 2.0 \
--num_workers 8 \
--devices 1 \
2>&1 |tee -a ${savepath}/log.txt |