Профилирование времени выполнения ResNet

Я использовал набор данных CIFAR-10 для обучения и оценки ResNet на процессоре Intel i7. (Модель ResNet находится в Tensorflow: https://github.com/tensorflow/models/tree/master/official/resnet)

Теперь меня интересует профилирование приложения, то есть определение времени выполнения основных функций. Анализ с помощью функций sort_stats() дает информацию только о верхней функции. Более того, использование инструмента профилирования cProfile выдает следующую ошибку:

профиль python3 -m cifar10_main.py

Файл "/usr/local/lib/python3.5/dist-packages/tensorflow/python/platform/app.py", строка 124, при запуске _sys.exit(main(argv)) TypeError: main() принимает 0 позиционных аргументы, но 1 был дан

Было бы здорово, если бы кто-нибудь помог мне собрать (почти точную и с функциональным или линейным уровнем) информацию о профилировании для ResNets. Спасибо :)


person Shail Dave    schedule 26.12.2017    source источник


Ответы (1)


Я бы использовал tf.profiler. Если вы не выполняете с энтузиазмом, интересные проблемы с производительностью (после построения графика) будут в коде TensorFlow C++, а не в Python.

person Allen Lavoie    schedule 02.01.2018