From 6f6714fcc0f96d5031acde3c47cf81c6e34fcdab Mon Sep 17 00:00:00 2001
From: Wenjie Du <wenjay.du@gmail.com>
Date: Tue, 3 Sep 2024 16:57:59 +0800
Subject: [PATCH] test: make the TimeMixer smaller for a faster test;

---
 tests/imputation/timemixer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/imputation/timemixer.py b/tests/imputation/timemixer.py
index bf5c72cd..a0735663 100644
--- 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,