模型压缩之神经网络模型参数量统计工具thop
神经网络参数量统计工具thop话不多说,直接上代码。import torchfrom torchvision import modelsfrom thop import profilemodel = models.densenet121()input = torch.randn(1, 3, 224, 224)flops, params = profile(model, inputs...
原创
2019-07-18 14:59:44 ·
1810 阅读 ·
1 评论