Хабр Курсы для всех
РЕКЛАМА
Большая витрина: от крупнейших школ до частных авторов. Сравнивайте по цене, длительности, формату и выбирайте самый подходящий курс!
А для Tensorflow есть средства визуального программирования вроде майкрософтовского Machine Learning Studio для Azure ML?
total += BATCH_SIZE не вполне корректно. Один из возможных вариантов — среди прочего в sess.run считать что-то вроде tf.shape(img_batch)[0].Верное замечаение.
Вообще правильнее решить задачу было бы с помощью tf.metrix.accuarcy, но в версии 1.4 статистику нельзя сбрасывать между эпохами, что не очень удобно.
Асинхронная загрузка больших датасетов в Tensorflow