Классификация патчей с большого изображения с использованием pytorch

Вы можете поймать конкретную ошибку, которая изящно закончит обучение, когда все примеры будут исчерпаны:

try:
    while True:
        # Run training Ops here...

except tf.errors.OutOfRangeError:
    print('Done training -- epoch limit reached')