【Deep Learning】计算PyTorch模型的FLOPs

1. 衡量深度学习模型重量级的指标 参数量: CNN: 一个卷积核的参数 = kkCin+1 一个卷积层的参数 = (一个卷积核的参数)卷积核数目=kkCinCout+Cout FLOPS: 注意全大写,是floating point operations per second的缩写,意指每秒浮点运算次数,理解为计算速度。是一个衡量硬件性能的指标。 FLOPs: 注意s小写,是floating p
相关文章
相关标签/搜索