Я занят, пытаясь профилировать код BERT тензорного потока , чтобы посмотреть, достаточно ли он быстр для конкретного случая использования.
Мне нужно определить скорость предсказаний.Теперь, как я понимаю, можно добавить профилировщик или run_metadata объект к session.Run опциям.Тем не менее, BERT использует новый Оценщик API.
Как один профиль оценщиков?Это возможно?
Обновление:
Команда Tensorflow показала мне здесь , что можно присоединить ProfilerHook к оценщику:
hooks = [tf.train.ProfilerHook(save_steps=1, output_dir=model_dir)]
estimator.predict(..., hooks=hooks)
Создает файл трассировки, но трассировка не работает.
Error : Couldn't create an importer for the provided eventData.
at Import.createImporter_ (chrome://tracing/tracing.js:1300:2071)
at addImportStage (chrome://tracing/tracing.js:1295:167)
at Task.run (chrome://tracing/tracing.js:2307:95)
at runAnother (chrome://tracing/tracing.js:2310:371)
at runTask (chrome://tracing/tracing.js:2286:57)
at processIdleWork (chrome://tracing/tracing.js:2291:116)
at window.requestIdleCallback.timeout (chrome://tracing/tracing.js:2284:81)