- torchsummary를 쓸 때 model의 forward 부분 인자가 여러 개인 경우
- summary(model, [input1 size, input2 size, ...])
- input size에 배치 사이즈는 제외
- 리스트로 묶어주기
에러 발생!
AttributeError: 'list' object has no attribute 'size'
- torchsummaryX를 쓰면 해결 가능하다 함
- 배치 사이즈 포함한 tensor 데이터 만들어주기
- 인자로 넣을 때 리스트화 필요없이 그대로 넣어주기
