Python 클래스 피킹 방법 TensorFlow 그래프?

Question

필자는 동적 재현 가능 보고서를 만들기 위해 pythontex를 사용합니다. tensorflow 그래프를 래핑하는 클래스가 있습니다.

내 문서에서 클래스를 만들고 거기에 tensorflow 그래프를 구성합니다.

텐서 흐름 모듈을 가져오고 그래프를 작성하는 것은 긴 과정이므로 매번이 프로세스를 다시 실행하지 않도록 결과를 피킹하고 싶습니다.

하지만 오류가 발생합니다. can't pickle class tensorflow.python.util.xxx. 나는 그것이 알려진 문제라고 생각합니다. 클래스를 사용자 정의하여 편리하게 덤프하고로드 할 수 있습니까? 아니면 파이썬 세션이나 tensorflow 세션을 피클 할 수 있을까요?

Answer 1

tf.train.Saver을 사용해보세요. 누군가 here에 대한 유용한 게시물을 작성했습니다. 모듈의 최상위 레벨에 정의 된 클래스 만 피클 할 수 있습니다 (tensorflow.xxx은 괜찮지 만 tensorflow.python.util.xxx은 그렇지 않습니다). 따라서 피클을 피할 수 없습니다 (피클 링에 대한 자세한 내용은 here 참조). 가장 가능성있는 해결책은 파일에 저장하는 다른 방법을 찾는 것입니다.