From f7034c3917c0a865b37b48fea9b41d00ffb6febd Mon Sep 17 00:00:00 2001
From: MingzheYang <zhepro88@gmail.com>
Date: Wed, 28 Jan 2026 01:14:20 +0800
Subject: [PATCH] Reduce batch size and seq length for CUDA memory

---
 example/config.json | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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