diff --git a/train_configs/minigptv2_finetune.yaml b/train_configs/minigptv2_finetune.yaml index 114d7e9..383f148 100644 --- a/train_configs/minigptv2_finetune.yaml +++ b/train_configs/minigptv2_finetune.yaml @@ -267,7 +267,7 @@ run: # optimizer lr_sched: "linear_warmup_cosine_lr" init_lr: 1e-5 - min_lr: 8e-5 + min_lr: 1e-6 warmup_lr: 1e-6 weight_decay: 0.05 @@ -291,4 +291,4 @@ run: distributed: True wandb_log: True - job_name: minigptv2_finetune \ No newline at end of file + job_name: minigptv2_finetune