add grid search

This commit is contained in:
thanhvc3 2023-05-17 13:59:24 +07:00
parent e7e42739e2
commit ddbaa2781f

View File

@ -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,