typo fix

keras-team · May 27, 2020 · 6d57bf0 · 6d57bf0
1 parent 95f2880
commit 6d57bf0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autokeras/engine/tuner.py b/autokeras/engine/tuner.py
@@ -48,7 +48,7 @@ def get_best_model(self):
         model.load_weights(self.best_model_path)
         return model
 
-    def _on_train_end(self, model, hp, x, *args, **kwargs):
+    def _on_train_begin(self, model, hp, x, *args, **kwargs):
         """Adapt the preprocessing layers and tune the fit arguments."""
         self.adapt(model, x)