fix config example in pytorch mlp documentation

This commit is contained in:
Yinon Polak 2023-03-28 14:41:25 +03:00
parent 026b6a39a9
commit b795a70102
2 changed files with 4 additions and 4 deletions

View File

@ -26,7 +26,7 @@ class PyTorchMLPClassifier(BasePyTorchClassifier):
"trainer_kwargs": {
"max_iters": 5000,
"batch_size": 64,
"max_n_eval_batches": None,
"max_n_eval_batches": null,
},
"model_kwargs": {
"hidden_dim": 512,

View File

@ -27,7 +27,7 @@ class PyTorchMLPRegressor(BasePyTorchRegressor):
"trainer_kwargs": {
"max_iters": 5000,
"batch_size": 64,
"max_n_eval_batches": None,
"max_n_eval_batches": null,
},
"model_kwargs": {
"hidden_dim": 512,