크로스엔트로피 == 소프트맥스
7주차 과제를 GPU로 돌리고 잘못인식된 이미지를 확인하려고 하니 에러가 났다. 그래서 모델을 다시 CPU에 할당하고 확인했다.
model.to("cpu")