chore: Increase per_device_batch_size to 64 for roest-809m

alexandrainst · Oct 23, 2024 · 9f13a9d · 9f13a9d
1 parent 559354b
commit 9f13a9d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/makefile b/makefile
@@ -146,7 +146,7 @@ roest-809m:  ## Train the Røst-809M model
 		dataloader_num_workers=4 \
 		model_id=roest-809m \
 		private=true \
-		per_device_batch_size=32
+		per_device_batch_size=64
 
 roest-1b:  ## Train the Røst-1B model
 	@accelerate launch \