2017-03-27 3 views
1

Tensorflow (Python API)를 사용하여 처음부터 구현하고 훈련 한 CNN의 무게와 편향을 저장하고 싶습니다.CNN의 무게를 재사용하기 위해 가장 좋은 방법

이제 이러한 가중치를 파일에 저장하고 누군가와 공유하여 내 네트워크를 사용할 수 있습니다. 그러나 나는 많은 무게를 가지고 있기 때문에 나는 모른다. 어떻게 할 수 있습니까? 이를 수행 할 수있는 형식이 있습니까?

+0

Tensorflow는 모델을 저장하는 방법을 제공합니다 (https://www.tensorflow.org/api_docs/python/tf/train/Saver). 친구는 Tensorflow를 사용하여로드해야합니다. – mprat

+0

감사합니다! 이러한 매개 변수를 C++ API와 함께 사용하려면 작동합니까? –

+0

예, 언어 API가 다른 동일한 패키지의 저장 형식에는 차이가 없습니다. – mprat

답변

0

Tensorflow는 모델을 저장하는 방법을 제공합니다 : tensorflow.org/api_docs/python/tf/train/Saver. 친구는 Tensorflow를 사용하여로드해야합니다. 로드/저장 언어는 저장 방법에 영향을 미치지 않습니다. Python의 Tensorflow에 저장하면 C++의 Tensorflow에서 읽을 수 있습니다.