Skip to content

Commit

Permalink
fix(automl/tuning): default batch size to 32
Browse files Browse the repository at this point in the history
  • Loading branch information
ruancomelli committed Jun 27, 2022
1 parent 13c7323 commit 07154f2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion boiling_learning/automl/tuning.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ def fit_hypermodel(
ds_train,
validation_data=ds_val,
callbacks=params.callbacks,
batch_size=params.batch_size,
batch_size=params.batch_size or 32,
)

return ModelArchitecture(automodel.export_model())

0 comments on commit 07154f2

Please sign in to comment.