From ddbaa2781fcbb53def089f5dd7542fd2106f7ff9 Mon Sep 17 00:00:00 2001
From: thanhvc3 <thanhvc3@vng.com.vn>
Date: Wed, 17 May 2023 13:59:24 +0700
Subject: [PATCH] add grid search

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 31f024b..625b6e8 100644
--- a/main.py
+++ b/main.py
@@ -641,7 +641,7 @@ if __name__ == "__main__":
         from skorch import NeuralNet
 
         estimator = NeuralNet(
-            module=FouriER,
+            module=FouriER(model.p),
             criterion=torch.nn.BCELoss,
             optimizer=torch.optim.Adam,
             max_epochs=100,