update

2026-01-22 21:02:56 +08:00
parent 9ba96fe77e
commit 25a3647b02
2 changed files with 9 additions and 5 deletions
--- a/example/config.json
+++ b/example/config.json
@@ -7,10 +7,10 @@
  "out_dir": "./results",
  "device": "auto",
  "timesteps": 400,
-  "batch_size": 64,
-  "seq_len": 128,
-  "epochs": 5,
-  "max_batches": 2000,
+  "batch_size": 256,
+  "seq_len": 256,
+  "epochs": 12,
+  "max_batches": 8000,
  "lambda": 0.5,
  "lr": 0.0005,
  "seed": 1337,
@@ -26,5 +26,5 @@
  "use_tanh_eps": true,
  "eps_scale": 1.0,
  "sample_batch_size": 8,
-  "sample_seq_len": 128
+  "sample_seq_len": 256
 }