如何估算模型训练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
能看不尽景,始是不凡人
posted @ 2021-12-30 08:43