Add flops calculation code

ChenFengYe · Mar 6, 2023 · 7e5b2d8 · 7e5b2d8
1 parent 3590786
commit 7e5b2d8
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/scripts/flops.py b/scripts/flops.py
@@ -120,8 +120,7 @@ def main():
         # flops = FlopCountAnalysis(model, batch).total()
         # print(flops)
         # flops_lst.append(flops)
-        print('macs', macs/1e9)
-        print('flops', macs*2/1e9)
+        print('macs', macs/1e9, 'G')
         return
         macs_lst.append(macs)
         params_lst.append(params)