From 2b001e8932b4893f8b1dc1a1e2498f997618c865 Mon Sep 17 00:00:00 2001 From: zanussbaum Date: Fri, 7 Apr 2023 17:41:45 -0400 Subject: [PATCH] fix: batch size --- configs/train/finetune_gptj.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/train/finetune_gptj.yaml b/configs/train/finetune_gptj.yaml index aa6bf4a8..ce6feef7 100644 --- a/configs/train/finetune_gptj.yaml +++ b/configs/train/finetune_gptj.yaml @@ -9,7 +9,7 @@ streaming: false num_proc: 64 dataset_path: "nomic-ai/turbo-500k-multi" max_length: 1024 -batch_size: 16 +batch_size: 8 # train dynamics lr: 2.0e-5