chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-04 01:50:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update minigptv2_finetune.yaml

Browse Source
This commit is contained in:
Jun Chen 2024-01-04 03:35:53 +08:00 committed by GitHub
parent d00e899c1f
commit 71df7647b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train_configs/minigptv2_finetune.yaml
View File

@ -267,7 +267,7 @@ run:
# optimizer # optimizer
lr_sched: "linear_warmup_cosine_lr" lr_sched: "linear_warmup_cosine_lr"
init_lr: 1e-5 init_lr: 1e-5
min_lr: 8e-5 min_lr: 1e-6
warmup_lr: 1e-6 warmup_lr: 1e-6
weight_decay: 0.05 weight_decay: 0.05
@ -291,4 +291,4 @@ run:
distributed: True distributed: True
wandb_log: True wandb_log: True
job_name: minigptv2_finetune job_name: minigptv2_finetune