• 카테고리

    질문 & 답변
  • 세부 분야

    컴퓨터 비전

  • 해결 여부

    미해결

파라미터 저장하는 소스 코드 에러

21.06.09 17:55 작성 조회수 277

0

안녕하세요. 수업 정말 잘 보고 있습니다.

파라미터를 저장하는 부분에서 계속 에러가 발생해서 질문을 올리게 됐습니다. 구글링을 해보아도 원인을 알 수 없었습니다.

제공해 주신 코드 내에서 이미 학습된 파라미터 있다면 그 파라미터를 이용해 테스트 데이터 셋의 정확도를 출력하고 프로그램을 종료 시키는 부분에서 print문 안의 compute_accuracy함수 때문에 에러가 발생하는 것 같은데...정확한 이유를 모르겠습니다.

print 부분에서 compute_accuracy 함수만을 삭제하고 실행시키면 정상 동작 합니다. 혹시 원인을 아신다면 답변 부탁드립니다. 

답변 1

답변을 작성해보세요.

0

안녕하세요~. 반갑습니다.

해당코드는 GPU 관련 문제인데요.

여러 원인이 있을수 있어서 우선 코드 상단에 아래 코드를 추가해서 tensorflow를 CPU 모드로 실행하시는것을 추천드립니다. (GPU 사용을 중지하는 코드입니다.)

import os

os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"   

os.environ["CUDA_VISIBLE_DEVICES"]="-1"

감사합니다.