Merge pull request #87 from TanayNarshana/smaller-network-fix

Fix Mac count displayed for small networks.
sovrasov · Feb 7, 2022 · 5f2a45f · 5f2a45f
2 parents bb9cad5 + e491a9f
commit 5f2a45f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ptflops/utils.py b/ptflops/utils.py
@@ -7,7 +7,7 @@
 '''
 
 
-def flops_to_string(flops, units='GMac', precision=2):
+def flops_to_string(flops, units=None, precision=2):
     if units is None:
         if flops // 10**9 > 0:
             return str(round(flops / 10.**9, precision)) + ' GMac'