fix: smaller bs for 40gb

This commit is contained in:
Zach Nussbaum 2023-04-08 21:36:20 +00:00
parent 7f95ab3a06
commit 2f0eba211d

View File

@ -9,7 +9,7 @@ streaming: false
num_proc: 64
dataset_path: "nomic-ai/turbo-500k-multi"
max_length: 1024
batch_size: 4
batch_size: 2
# train dynamics
lr: 2.0e-5