记录torchsummary报AttributeError: 'xxx' object has no attribute 'size'错误:
选择合适的torchsummary版本即可。
pip uninstall torchsummary (version==1.5.1)
pip install torchsummary==1.4.4
记录torchsummary报AttributeError: 'xxx' object has no attribute 'size'错误:
选择合适的torchsummary版本即可。
pip uninstall torchsummary (version==1.5.1)
pip install torchsummary==1.4.4