-
카테고리
-
세부 분야
컴퓨터 비전
-
해결 여부
미해결
파라미터 저장하는 소스 코드 에러
21.06.09 17:55 작성 조회수 277
0
안녕하세요. 수업 정말 잘 보고 있습니다.
파라미터를 저장하는 부분에서 계속 에러가 발생해서 질문을 올리게 됐습니다. 구글링을 해보아도 원인을 알 수 없었습니다.
제공해 주신 코드 내에서 이미 학습된 파라미터 있다면 그 파라미터를 이용해 테스트 데이터 셋의 정확도를 출력하고 프로그램을 종료 시키는 부분에서 print문 안의 compute_accuracy함수 때문에 에러가 발생하는 것 같은데...정확한 이유를 모르겠습니다.
print 부분에서 compute_accuracy 함수만을 삭제하고 실행시키면 정상 동작 합니다. 혹시 원인을 아신다면 답변 부탁드립니다.
차량 번호판 인식 프로젝트와 TensorFlow로 배우는 딥러닝 영상인식 올인원
tf.train.CheckpointManager API를 이용해서 파라미터 저장하고 불러오기
강의실 바로가기
답변을 작성해보세요.
0
AISchool
지식공유자2021.06.10
안녕하세요~. 반갑습니다.
해당코드는 GPU 관련 문제인데요.
여러 원인이 있을수 있어서 우선 코드 상단에 아래 코드를 추가해서 tensorflow를 CPU 모드로 실행하시는것을 추천드립니다. (GPU 사용을 중지하는 코드입니다.)
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="-1"
감사합니다.
답변 1