fix arg in submission_runner

mlcommons · Nov 28, 2023 · 6885e69 · 6885e69
1 parent 25b3fbe
commit 6885e69
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/submission_runner.py b/submission_runner.py
@@ -228,7 +228,7 @@ def train_once(
       ]
       eager_backend_workloads = ['librispeech_deepspeech']
       aot_eager_backend_workloads = []
-      base_workload = workloads.get_base_workload_name(FLAGS.workload)
+      base_workload = workloads.get_base_workload_name(workload)
       if base_workload in compile_error_workloads:
         logging.warning(
             'These workloads cannot be fully compiled under current '