足迹

能看不尽景,始是不凡人

 

2021年12月30日

如何估算模型训练T(FL)OPS efficiency

摘要: Naive方法 以Torch Vision ResNet50-v1.5为例。 Step 1: 获取模型的前向理论需求MACs(Multiply–ACcumulate) 可使用thop得到模型的前向MACS。使用如下代码可得Torch Vision ResNet50-v1.5的前向MACs为4.112 阅读全文

posted @ 2021-12-30 08:43 姚伟峰 阅读(279) 评论(0) 推荐(0) 编辑

导航