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