test coverage

Author: haifeng-jin

autokeras/graph.py

@@ -296,7 +296,9 @@ def _compile_keras_model(self, hp, model):
         elif optimizer_name == "sgd":
             optimizer = keras.optimizers.SGD(learning_rate=learning_rate)
         elif optimizer_name == "adam_weight_decay":
-            steps_per_epoch = max(1, int(self.num_samples / 32))
+            steps_per_epoch = max(
+                1, int(self.num_samples / (self.batch_size or 32))
+            )
             num_train_steps = steps_per_epoch * self.epochs
 
             lr_schedule = keras.optimizers.schedules.PolynomialDecay(