pytorch解微分方程_【他山之石】pytorch计算模型FLOPs和Params

60 阅读 0 评论 40 点赞

我是靠谱客的博主内向吐司，最近开发中收集的这篇文章主要介绍pytorch解微分方程_【他山之石】pytorch计算模型FLOPs和Params，觉得挺不错的，现在分享给大家，希望可以做个参考。

“他山之石，可以攻玉”，站在巨人的肩膀才能看得更高，走得更远。在科研的道路上，更需借助东风才能更快前行。为此，我们特别搜集整理了一些实用的代码链接，数据集，软件，编程技巧等，开辟“他山之石”专栏，助你乘风破浪，一路奋勇向前，敬请关注。

作者：知乎—庄六岁

地址：https://www.zhihu.com/people/zhuang-ming-xi-31-16

网络框架模型计算量影响到模型的推断时间，模型的参数量对设备内存有要求，为了进行模型比较给大家介绍两种计算FLOPs和Params实用的小工具：

第一种方法 https://github.com/Lyken17/pytorch-OpCounter 安装方法：

pip install thop(推荐用这个) 或者 pip install --upgrade git+https://github.com/Lyken17/pytorch-OpCounter.git(这个方法需要同时安装pytorch)

用法：

from torchvision.models import resnet50from thop import profilemod

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。