인프런 커뮤니티 질문&답변
작성한 질문수
파라미터 저장하는 소스 코드 에러
작성
·
418
0
안녕하세요. 수업 정말 잘 보고 있습니다.
파라미터를 저장하는 부분에서 계속 에러가 발생해서 질문을 올리게 됐습니다. 구글링을 해보아도 원인을 알 수 없었습니다.
제공해 주신 코드 내에서 이미 학습된 파라미터 있다면 그 파라미터를 이용해 테스트 데이터 셋의 정확도를 출력하고 프로그램을 종료 시키는 부분에서 print문 안의 compute_accuracy함수 때문에 에러가 발생하는 것 같은데...정확한 이유를 모르겠습니다.
print 부분에서 compute_accuracy 함수만을 삭제하고 실행시키면 정상 동작 합니다. 혹시 원인을 아신다면 답변 부탁드립니다.
퀴즈
69%나 틀려요. 한번 도전해보세요!
딥러닝 모델 학습 중 매개변수를 주기적으로 저장하는 주된 이유는 무엇일까요?
모델 크기를 줄여 메모리 사용량을 줄이기 위해
과적합(Overfitting) 발생 여부를 쉽게 확인하기 위해
학습 중간에 문제가 발생해도 중단된 지점부터 재개하거나 학습된 모델을 활용하기 위해
모델 구조를 변경하더라도 이전 매개변수를 유지하기 위해
답변 1
0
AISchool
지식공유자
안녕하세요~. 반갑습니다.
해당코드는 GPU 관련 문제인데요.
여러 원인이 있을수 있어서 우선 코드 상단에 아래 코드를 추가해서 tensorflow를 CPU 모드로 실행하시는것을 추천드립니다. (GPU 사용을 중지하는 코드입니다.)
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="-1"
감사합니다.