Reduce batch size and seq length for CUDA memory

2026-01-28 01:14:20 +08:00
parent 379c411153
commit f7034c3917
1 changed files with 2 additions and 2 deletions
--- 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,