Expose TimesNet internal dimension

2026-07-09 16:22:34 +08:00
parent 8078fcb835
commit 552e09aa01
3 changed files with 13 additions and 5 deletions
--- a/README.md
+++ b/README.md
@@ -89,7 +89,8 @@ The end-to-end next-step trainer supports the same DDP launch pattern:
 ```bash
 torchrun --standalone --nproc_per_node=4 train_next_step.py \
  --exposure_cache_dir ukb_exposure_cache \
-  --batch_size 128
+  --batch_size 128 \
+  --d_model 64
 ```
 Training-channel statistics are cached at
 `<exposure_cache_dir>/train_channel_stats.npz`; use