fix: smaller bs for 40gb

This commit is contained in:
Zach Nussbaum
2023-04-08 21:36:20 +00:00
parent f5ca8bedcf
commit 2827693a57

View File

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