Как показывает вопрос, если
E = number of epochs
B = batch_size
N = number of total images
Тогда это должно удовлетворять:
N = E*B