From 7cf7f865dfc4e2b98bca3624d5fc6f66ba71aefd Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Mon, 25 Sep 2023 20:02:28 +0000 Subject: [PATCH] turn off torch compile for conformer --- submission_runner.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/submission_runner.py b/submission_runner.py index 83130d4a9..b9ec99067 100644 --- 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: