torchsummary.summary 这是一个显示模型参数的方法,注意不是类,不需要创建对象,第一个参数是模型,即nn.module类,第二个是输入数据的shape,类型为元组,注意,不需要输入batch数, summary(model,input_data=(3,416,416)) 输入是三通道图像 如果是输入下面的就是错误的形式 summary(model,input_data=(batch_size,3,416,416))