tf.Estimator와 함께 Tensorflow 프로파일 러 사용

Question

Tensorflow 프로파일 러를 사용하여 어떤 이유로 천천히 실행되는 일부 코드의 프로파일을 작성해야합니다. 불행하게도, 문제의 코드는 tf.Estimator를 사용하므로 실행 메타 데이터 객체를 세션 run() 호출에 삽입하여 프로파일 러에 필요한 정보를 얻는 방법을 알 수 없습니다.

어떻게해야합니까?

Answer 1

설명 된대로 ProfileContext를 사용하십시오 (https://github.com/tensorflow/tensorflow/tree/master/tensorflow/core/profiler). 이렇게하면 세션에 액세스하지 않고도 프로파일 링 할 수 있습니다.