RuntimeError: Given groups=1, weight of size [400, 256, 1, 1], expected input[64, 3, 1, 1] to have 256 channels, but got 3 channels instead

반응형

# 에러 내용

RuntimeError: Given groups=1, weight of size [400, 256, 1, 1], expected input[64, 3, 1, 1] to have 256 channels, but got 3 channels instead

# 원인

nn.DataParallel 을 사용할 수 없음 ( torch 1.5-1.7 )

 

# 해결방안

torch 1.2로 downgrade