作者您好，我算出来paraTransCNN的参数量和论文里写的42.7差距有点大。请问您的模型参数量是怎么计算的

[DanpengDing]

if name == 'main':
net = ParaTransCNNCBAM().cuda()
input = torch.rand(1, 3, 224, 224).cuda()
output = net(input)
total = sum([param.nelement() for param in net.parameters()])
print("Number of parameter: %.2fM" % (total / 1e6))
这是我计算参数量的方法，在3060和TiTan上都算过和论文差距有点大