RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 16 but got size 15 for tensor number 1 in the list. 에러 해결

 

Unet 모델을 제작해서 돌리는데, 아래와 같은 에러가 나왔습니다. 정말 여기저기 찾아봤는데,, 해결이 안되었는데,

 

코드를 이것저것 돌려보니 해결이 되었네요..

 

RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 16 but got size 15 for tensor number 1 in the list.

 

위와 같은 에러가 발생할 경우

 

모델을 제작할때, 256으로 제작했을때, 실제 input으로도 같은 사이즈를 넣어줘야 합니다. 아니면 위와 같은 에러가 발생할수 있습니다.

 

 그렇기 때문에 transforms function을 이용해서 그림의 크기를 조절하는 Resize를 이용하여 분석하면 됩니다.

 

# inputs의 경우 모델 돌릴때 들어가는 input
augm = transforms.Compose([transforms.Resize(size=512)])
inputs = augm(inputs)

 

감사합니다.