How to estimate the maximum allowed batch size for a TensorFlow training?

You can estimate the largest batch size using:

Max batch size= available GPU memory bytes / 4 / (size of tensors + trainable parameters)

stackoverflow.com/a/46656508

See also: