[torch error] off-the-shelf 모델을 이식했는데 pretrained model 이 잘 동작하지 않을때

off-the-shelf 모델을 내 프로젝트에 이식했는데 돌아가기는 하지만 이식된 모델이 이상한 output 을 내보낼때가 있다.

이 때, 내가 무슨 실수를 한건지를 빠르게 되짚기 위해 디버깅 매뉴얼을 남기고자 한다.

 

1. 모델이 pretrained weight 를 잘 load 하는지 확인해본다.
2. 이미지가 제대로 load 되고 normalized image 를 입력하고 있는지 확인한다. 그리고 이 normalizing 방법은 off-the-shelf 모델이 학습에서 사용된 방법대로 적용되어야 한다.
3. model.eval() 했는지 확인한다.
4. 사실 모델은 잘 동작했는데 내가 시각화를 잘못한게 아닌지 확인한다. 예를 들어 출력된 mesh를 렌더링해서 확인하는 경우가 있다. 이럴 때 vertex 나 triangle faces 를 잘못 입력한 것은 아닌지 확인할 필요가 있다.