[ONNX] 결과 차이 클때, 의심해볼만한 것 inplace 연산

x -= .5 

x /= .5

 

위 처럼 코드 짰으면

 

아래 처럼 수정하자

 

x = x-.5

x = x/.5

 

torchvision.transforms.Normalize 를 모델에 집어넣었다가 변환시 토치 모델과 크게 상이한 값을 반환하는 문제가 발생했다.

 

이또한 torchvision.transforms.Normalize 의 함수 파라미터인 inplace 를 False 로 주니 해결됐다.

 

torchvision.transforms.Normalize(..., inplace=False)