diff --git a/example/config.json b/example/config.json
index 9258792..8d7866d 100644
--- a/example/config.json
+++ b/example/config.json
@@ -7,7 +7,7 @@
   "out_dir": "./results",
   "device": "auto",
   "timesteps": 600,
-  "batch_size": 128,
+  "batch_size": 32,
   "seq_len": 128,
   "epochs": 10,
   "max_batches": 4000,