flops counter formatting fix (#3837)

nni/compression/pytorch/utils/counter.py

@@ -313,12 +313,13 @@ class ModelProfiler:
 
         table.field_names = headers
         for i, result in enumerate(self.results):
+            flops_count = int(result['flops'].item()) if isinstance(result['flops'], torch.Tensor) else int(result['flops'])
             row_values = [
                 i,
                 result['name'],
                 result['module_type'],
                 str(result['weight_shape']),
-                result['flops'],
+                flops_count,
                 result['params'],
             ]
             name_counter[result['name']] += 1
@@ -408,4 +409,4 @@ def count_flops_params(model, x, custom_ops=None, verbose=True, mode='default'):
         print(f'FLOPs total: {profiler.sum_flops()}')
         print(f'#Params total: {profiler.sum_params()}')
 
-    return profiler.sum_flops(), profiler.sum_params(), profiler.results
\ No newline at end of file
+    return profiler.sum_flops(), profiler.sum_params(), profiler.results