Hello, Can I specify batch_size as one of the arguments in model.fit() method? Would that work same way as predetermining batch_size using tf.batch method? Thanks.

Hi, Specifying the "batch_size" parameter in the fit method works for inputs such as numpy array. In the MNIST example we are working with dataset object, that's why we are separately shuffling and batching it. When working with NumPy arrays, the batch size in the fit method automatically shuffles the dataset and batch it. Best, 365 Team

Follow this topic

Ahmed Mady

Last answered:

18 Feb 2021

Posted on:

16 Feb 2021

Resolved: Specifying batch_size in many ways

in Convolutional Neural Networks with TensorFlow in Python / Building and training the CNN

1 answers ( 1 marked as helpful)

Nikola Pulev

Instructor

Posted on:

18 Feb 2021

Helpful answer

Resolved: Specifying batch_size in many ways

Submit an answer

related questions