[Error] TypeError: conv2d(): argument 'input' (position 1) must be Tensor, not NoneType

TypeError: conv2d(): argument 'input' (position 1) must be Tensor, not NoneType

 

원인 : 

nn.Conv2d()에 input type으로 Tensor type이 들어가야 하는데 NoneType이 들어가고 있다는 말이다. 왜 NoneType이 들어가고 있는지 코드에서 확인해보아야 한다.

 

에러가 발생한 곳

 

layer2의 정의

 

self.layer2의 정의를 살펴보니 ResidualBlock 클래스의 내부를 보아야 했다.

 

ResidualBlock class의 forward()

 

ResidualBlock class의 forward이다. 문제를 확인할 수 있겠는가?

forward()의 return 을 설정해두지 않아서 NoneType 에러가 발생한 것이다.

 

해결법 : forward()에 return을 달아두자...