add grid search

2023-05-17 13:59:24 +07:00
parent e7e42739e2
commit ddbaa2781f
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,