Add new models to report (#1475)

Co-authored-by: Mitchelle Rasquinha <80070689+mrasquinha-g@users.noreply.github.com>
mlcommons · Aug 15, 2023 · 62a1fea · 62a1fea
1 parent 127b761
commit 62a1fea
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tools/submission/generate_final_report.py b/tools/submission/generate_final_report.py
@@ -103,7 +103,8 @@ def main():
   columns_order = [['Result'],
                    [
                        'resnet', 'retinanet', '3d-unet-99', '3d-unet-99.9',
-                       'rnnt', 'bert-99', 'bert-99.9', 'dlrm-99', 'dlrm-99.9'
+                       'rnnt', 'bert-99', 'bert-99.9', 'dlrm-v2-99', 'dlrm-v2-99.9',
+                       'gptj-99', 'gptj-99.9'
                    ], ['SingleStream', 'MultiStream', 'Server', 'Offline'],
                    [
                        'Latency (ms)',
@@ -120,8 +121,8 @@ def main():
           'rnnt': ['Server', 'Offline'],
           'bert-99': ['Server', 'Offline'],
           'bert-99.9': ['Server', 'Offline'],
-          'dlrm-99': ['Server', 'Offline'],
-          'dlrm-99.9': ['Server', 'Offline'],
+          'dlrm-v2-99': ['Server', 'Offline'],
+          'dlrm-v2-99.9': ['Server', 'Offline'],
           '3d-unet-99': ['Offline'],
           '3d-unet-99.9': ['Offline'],
           'gptj-99': ['Server', 'Offline'],