[TIL]Day 58

이재희·2021년 1월 26일
0

TIL

목록 보기
58/312

크로스엔트로피 == 소프트맥스

7주차 과제를 GPU로 돌리고 잘못인식된 이미지를 확인하려고 하니 에러가 났다. 그래서 모델을 다시 CPU에 할당하고 확인했다.

model.to("cpu")

profile
오늘부터 열심히 산다

0개의 댓글