add grid search

2023-05-17 13:59:24 +07:00 · 2023-05-17 13:59:24 +07:00 · ddbaa2781f
commit ddbaa2781f
parent e7e42739e2
1 changed files with 1 additions and 1 deletions
--- 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,