fix: add wd + min lr to config

pull/913/head
Zach Nussbaum 1 year ago
parent c0a9065032
commit 5de765410a

@ -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

Loading…
Cancel
Save