Tensorflow Training Слои вычисление времени - PullRequest
0 голосов
/ 29 мая 2018

Я пытаюсь получить время вычислений различных слоев из шага тренировки, мне удалось получить информацию о шагах тренировки с опциями run_meta_data, я могу увидеть время вычисления в Tensorboard каждого слоя.я проанализировал метаданные для json, создав объект временной шкалы.а также я преобразовал протобуф внутри событий, хранящихся в тензорном потоке.в обоих случаях, когда я вычисляю время слоя (суммируя длительность узлов в файле protobuf или json), они отличаются от показанных на графике.я что-то пропустил?

Обновление protobuf анализируется в словаре.узлы описаны ниже.

{"имя_узла": "Tower_0 / conv1 / Conv2D: Conv2D", "opEndRelMicros": "585", "allEndRelMicros": "585", "allStartMicros": "1527330751389528"},
{"nodeName": "Tower_0 / conv1 / Conv2D: Conv2D", "opEndRelMicros": "1721", "allEndRelMicros": "1721", "allStartMicros": "1527330751390116"},
{"nodeName":" Tower_0 / conv1 / Conv2D: Conv2D "," opEndRelMicros ":" 10 "," allEndRelMicros ":" 10 "," allStartMicros ":" 1527330751391839 "},
Что я сделал, чтобы получить время,переберите весь файл, найдите узлы с именем, совпадающим с именем слоя, и добавьте время opeEndRelMicros.времена выше в некоторых слоях и ниже в некоторых слоях.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...