使用summary时遇到AttributeError: ‘list‘ object has no attribute ‘size‘的错误

出现改错时,意思就是本来因该是tensor的地方你写的是list,所以出现了这个错误。
对我个人而言,我的错误出现在模型的forword函数中,我的模型有多个输入,而我是以list的形式传入了多个输入,即将多个输入作为了一个整体传给了模型。而在summary中会计算模型的输入形状,即其会使用到输入的.size属性,而当我传入的是list时,由于list没有.size属性,所以报了这个错。正确的的做法是将这多个输入作为多个tensor分开传入,而不是作为一个list整体。如下:

    def forward(self, c0, c1, c2, c3, c4):

 

你可能感兴趣的:(python,tensorflow,算法,深度学习,pytorch)