test: make the TimeMixer smaller for a faster test;

WenjieDu · Sep 3, 2024 · 6f6714f · 6f6714f
1 parent 57d9789
commit 6f6714f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/imputation/timemixer.py b/tests/imputation/timemixer.py
@@ -47,8 +47,8 @@ class TestTimeMixer(unittest.TestCase):
         DATA["n_features"],
         n_layers=2,
         top_k=5,
-        d_model=512,
-        d_ffn=512,
+        d_model=32,
+        d_ffn=32,
         dropout=0.1,
         epochs=EPOCHS,
         saving_path=saving_path,