Tune Batch Size Keras Tuner