t push
This commit is contained in:
yiranyyu
2024-05-28 17:15:07 +08:00
2 changed files with 3 additions and 3 deletions

View File

@@ -34,8 +34,8 @@ torchrun $DISTRIBUTED_ARGS finetune.py \
--fp16_full_eval false \
--do_train \
--do_eval \
--tune_vision false \
--tune_llm false \
--tune_vision true \
--tune_llm true \
--model_max_length 2048 \
--max_steps 10000 \
--eval_steps 1000 \

View File

@@ -32,7 +32,7 @@ torchrun $DISTRIBUTED_ARGS finetune.py \
--bf16_full_eval true \
--do_train \
--do_eval \
--tune_vision false \
--tune_vision true \
--tune_llm false \
--use_lora true \
--lora_target_modules "llm\..*layers\.\d+\.self_attn\.(q_proj|k_proj)" \