728x90
반응형
weight 파일을 load 하지 않고 테스트 시
model 구조만 불러와서 inferece 하기 때문에
분류 결과는 class A : 0.54, class B : 0.46 이런식의 값이 산출되게 된다.
반드시 weight 파일을 load 할 때 path 를 잘 지정해주어야 하며, (예외 처리도 반드시 해야함)
load_state_dict(checkpoint['state_dict'] 을 꼭 해주어야한다.
728x90
반응형
'삽질 기록' 카테고리의 다른 글
| mmdetection 으로부터 학습한 모델을 ONNX 및 TensorRT 로 변환 시 나타나는 문제 (0) | 2021.07.26 |
|---|---|
| fatal error: libpq-fe.h: No such file or directory (0) | 2021.07.15 |
| Linux 환경에서 OpenCV의 VideoCapture 사용 시 Webcam이 죽어도 읽히지 않을 때 (0) | 2021.06.25 |
| Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.FloatTensor) should be the same (0) | 2021.06.24 |
| Ubuntu 18.04 환경에서 VNC Server 설정했을 때 회색 화면 나타나는 문제 (0) | 2021.06.11 |