Fix Mac count displayed for small networks.

sovrasov · Feb 2, 2022 · e491a9f · e491a9f
1 parent bb9cad5
commit e491a9f
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'