计算网络参数量和复杂度

from thop import profile

from model1.trans import Net

model = Net()
print(model)
print('model parameters:',sum(param.numel() for param in model.parameters()))
input =torch.randn(1,3,256,256)     #输入图像尺寸大小
flops,params = profile(model,inputs=(input,))
print('FLOPs = ' + str(flops/1000**3) + 'G')
print('Params = ' + str(params/1000**2) + 'M')

输出为如下图所示