TypeError: conv2d() received an incalid combination of arguments

TypeError: Conv2d() received an incalid combination of arguments

해당 오류는 Classification Task 진행하던 도중에 만나게 되었다. 

Module에 관련된 오류로 Conv2d 자체의 문제 보단 input으로 들어가는 데이터와 관련된 오류이다.

# 이미지 가시화 진행
return {'img': transformed, 'class': label}
# 학습 적용시
return transformed, label

따라서 visualize 진행할 경우 dict 자료 구조로 반환을 넘겨주면 되지만 학습시엔 label 자체를 넘겨줘야한다.