cngthnh/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add grid search

Browse Source
This commit is contained in:
thanhvc3 2023-05-17 13:59:24 +07:00
parent e7e42739e2
commit ddbaa2781f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@ -641,7 +641,7 @@ if __name__ == "__main__":
from skorch import NeuralNet from skorch import NeuralNet
estimator = NeuralNet( estimator = NeuralNet(
module=FouriER, module=FouriER(model.p),
criterion=torch.nn.BCELoss, criterion=torch.nn.BCELoss,
optimizer=torch.optim.Adam, optimizer=torch.optim.Adam,
max_epochs=100, max_epochs=100,