Update get_flops.py

druzhkov-paul · May 26, 2020 · e13a2a1 · e13a2a1
1 parent ce85416
commit e13a2a1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/get_flops.py b/tools/get_flops.py
@@ -53,8 +53,8 @@ def main():
 
     if args.out:
         out = list()
-        out.append({'key': 'size', 'displayName': 'Size', 'value': float(params.split(' ')[0]), 'unit': 'Mp'})
-        out.append({'key': 'complexity', 'displayName': 'Complexity', 'value': 2 * float(flops.split(' ')[0]),
+        out.append({'key': 'size', 'display_name': 'Size', 'value': float(params.split(' ')[0]), 'unit': 'Mp'})
+        out.append({'key': 'complexity', 'display_name': 'Complexity', 'value': 2 * float(flops.split(' ')[0]),
                     'unit': 'GFLOPs'})
         with open(args.out, 'w') as write_file:
             json.dump(out, write_file, indent=4)