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

[dmitrii_fediuk]

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

[dmitrii_fediuk]

See also:

• What is the optimal batch size for a TensorFlow training?
• How to check Nvidia GPU memory usage in Ubuntu 18.04?
• What does mean «train_config» → «batch_size» in TensorFlow?