[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에서 해결 방법을 찾았다. 해결방법은 의외로 간단하다. torchsummaryX를 사용하면 된다.  아래 코드와 같이 이용하면 리스트 형태로 전달해도 Summary를 확인 할 수 있다.

from torchsummaryX import summary
arch = summary(model, torch.rand((1, 3, 256, 256)))

 

---

참고문헌

 

AttributeError: 'list' object has no attribute 'size' · Issue #137 · qubvel/segmentation_models.pytorch