[pytorch] summary 'list' object has no attribute ' size' error

pytorch에서 keras처럼 summary를 확인하는 함수가 있다. 이때 가장 많이 사용하는 함수가 아래의 함수일 것이다. from torchsummary import summary 필자도 torchsummary를 사용하여 model의 summary를 확인했다. 그러나 Hrnet을 설계할 때 값이 리스트 형태로 전달하면서 다음과 같은 오류가 발생했다. File "\torchsummary\torchsummary.py", line 19, in hook summary[m_key]["input_shape"] = list(input[0].size()) AttributeError: 'list' object has no attribute 'size' 그래서 구글에서 검색 중 github에서 해결 방법을 찾았다. ..