![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
暄染落墨
这个作者很懒,什么都没留下…
展开
-
分配算法之匈牙利算法
参考链接:https://brc2.com/the-algorithm-workshop/n个工人n份工作,一一匹配,每份工作只能被一个工作完成,每个工人只能完成一份工作,暴力破解的话,复杂度是n!。一个n*n的有权值矩阵,1.找到每行的最小值,每行的元素减去这个这个最小值2.找到每列的最小值,每列的元素减去这个这个最小值重复以下3个步骤(3-5):3.用最少的行线和列线去覆盖全部的零4.判断是否满足退出循环条件(若步骤3中线的数量等于行或者列元素的个数,则退出,否则继续)5.找到矩阵中不在原创 2021-08-31 17:37:51 · 216 阅读 · 0 评论 -
参数量的计算(以resnet50为例,bn层的参数
目录模型参数量的计算一模型参数量的计算二resnet50的网络结构:模型参数量的计算一print("resnet50 have {} paramerters in total".format(sum(x.numel() for x in resnet50.parameters())))模型参数量的计算二num_params = 0for param in netG.parameters(): num_params += param.numel()print(num_params / 1e6)原创 2021-08-16 22:51:53 · 8928 阅读 · 0 评论 -
FLOPs和使用fvcore计算FLOPs
FLOPS:(floating point operations per second)浮点运算次数每秒,是计算速度,是衡量硬件性能的指标。FLOPs:(floating point operations)小写的s表复数的意思,是计算量,用来衡量算法/模型的复杂度。""" Provides access to per-submodule model flop count obtained by tracing a model with pytorch's jit tracing func原创 2021-08-06 18:41:52 · 1281 阅读 · 0 评论