|
#!/bin/bash |
|
|
|
dataset="iu_xray" |
|
annotation="data/iu_xray/annotation.json" |
|
base_dir="./data/iu_xray/images" |
|
|
|
version="v1_delta" |
|
savepath="./save/$dataset/$version" |
|
|
|
python -u train.py \ |
|
--dataset ${dataset} \ |
|
--annotation ${annotation} \ |
|
--base_dir ${base_dir} \ |
|
--batch_size 8 \ |
|
--val_batch_size 12 \ |
|
--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 2 \ |
|
--max_epochs 15 \ |
|
--limit_val_batches 1.0 \ |
|
--val_check_interval 1.0 \ |
|
--num_sanity_val_steps 2 \ |
|
2>&1 |tee -a ${savepath}/log.txt |
