如何估算模型训练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 姚伟峰 阅读(375) 评论(0) 推荐(0) 编辑