2021
03-05
03-05
pytorch 计算Parameter和FLOP的操作
深度学习中,模型训练完后,查看模型的参数量和浮点计算量,在此记录下:1THOP在pytorch中有现成的包thop用于计算参数数量和FLOP,首先安装thop:pipinstallthop注意安装thop时可能出现如下错误:解决方法:pipinstall--upgradegit+https://github.com/Lyken17/pytorch-OpCounter.git#下载源码安装使用方法如下:fromtorchvision.modelsimportresnet50#引入ResNet50模型fromthopimportprofilemodel=resnet50()fl...
继续阅读 >