2017-01-19 2 views
0

vgg-19 모델을 사용하여 스타일 전송을 구현하려고했습니다. 나는 텐데 플로우를 백엔드로 사용하여 vga-19의 기존 keras 모델을 통해 가중치를로드했습니다. 처음에는 노트북에서 시험해 보았지만 tf 0.11.0에서 매우 느리지 만 괜찮 았습니다. 그런 다음 버전 0.12.1 인 gpus로 AWS 인스턴스를 전환하고 동일한 스크립트를 시도했습니다. jupyter 노트북에서는 오류가 발생하지 않지만 대신 커널을 정지시키고 출력을 계산하려고 할 때 강제로 재시작합니다. 특정 레이어 (첫 번째 레이어의 입력 제외). 저는 0.11.0의 GPU 버전을 설치했습니다 - 두 버전 모두 행렬 곱셈의 테스트를 통과했지만 동일한 문제입니다.tensorflow가 python 콘솔에 로그가없는 sess.run()에서 충돌합니다.

또한 0.11.0 용으로 작성된 vgg-16을 사용하는 사람의 스크립트에서 0.12.1을 실행 해 보았습니다. 그리고 레이어 크기가 잘못되어 세션이 중단되었지만 CPU에서 0.11.0에서 정상적으로 실행되었습니다. tf 0.11.0의 GPU 버전 - 동결 및 재시작.

파이썬 콘솔의 경우를 제외하고는 tensorflow에서 작성한 로그가 있습니까? 나는 그 (것)들에 관하여 아무 언급도 찾아 낼 수 있지 않았다

답변

0

고쳐졌다. cudnn 버전이 tensorflow 버전과 일치하지 않습니다. 오류 출력은 jupiter 노트북이 아닌 python 콘솔에만 제공됩니다.

관련 문제