weixin_40690713的博客

原创 模型压缩之神经网络模型参数量统计工具thop

神经网络参数量统计工具thop 话不多说，直接上代码。 import torch from torchvision import models from thop import profile model = models.densenet121() input = torch.randn(1, 3, 224, 224) flops, params = profile(model, inputs...