Блог

Я тренирую нейронную сеть через TFRecordDataset. Однако в конце каждой эпохи, т. Е. С ETA: 0s , Обучение застревает на десятки секунд. Для справки, для завершения одной эпохи требуется около минуты для набора данных объемом около 25 ГБ (перед анализом подмножества функций).

Я запускаю TensorFlow 2.3.1 с графическим процессором Nvidia Titan RTX. Это предполагаемое поведение? Может быть, из-за предварительной обработки во входном конвейере? Выполняется ли эта предварительная обработка только процессором или выгружается на графический процессор? Спасибо!

Если у вас есть набор проверки, и вы используете model.fit() , вероятно, это время, необходимое для вычисления потерь и показателей. В большинстве случаев для вычисления показателей разделения 80/20 потребуется дополнительно 25%.