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