Archives/gpt4all
2
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all synced 2024-11-08 07:10:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: add wd + min lr to config

Browse Source
This commit is contained in:
Zach Nussbaum 2023-04-08 20:37:51 +00:00
parent c0a9065032
commit 5de765410a
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configs/train/finetune_lora.yaml
View File

@ -7,12 +7,14 @@ save_name: "nomic-ai/gpt4all-lora-multi-turn"
# dataset # dataset
streaming: false streaming: false
num_proc: 64 num_proc: 64
dataset_path: "data_multiturn" dataset_path: "nomic-ai/turbo-500k-multi"
max_length: 1024 max_length: 1024
batch_size: 4 batch_size: 4
# train dynamics # train dynamics
lr: 5.0e-5 lr: 5.0e-5
min_lr: 0
weight_decay: 0.0
eval_every: 2000 eval_every: 2000
eval_steps: 100 eval_steps: 100
save_every: 2000 save_every: 2000