turn off torch compile for conformer

mlcommons · Sep 25, 2023 · 7cf7f86 · 7cf7f86
1 parent faf775e
commit 7cf7f86
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/submission_runner.py b/submission_runner.py
@@ -209,9 +209,8 @@ def train_once(
     model_params, model_state = workload.init_model_fn(
         model_init_rng, dropout_rate, aux_dropout_rate)
     if FLAGS.framework == 'pytorch' and FLAGS.torch_compile:
-      compile_error_workloads = ['ogbg', 'criteo1tb']
-      eager_backend_workloads = [
-          'librispeech_conformer', 'librispeech_deepspeech'
+      compile_error_workloads = ['ogbg', 'criteo1tb', 'librispeech_conformer']
+      eager_backend_workloads = ['librispeech_deepspeech'
       ]
       aot_eager_backend_workloads = []
       if FLAGS.workload in compile_error_workloads: