diff --git a/submission_runner.py b/submission_runner.py index d7f28bc22..6577204f2 100644 --- a/submission_runner.py +++ b/submission_runner.py @@ -207,9 +207,7 @@ def train_once( model_init_rng, dropout_rate, aux_dropout_rate) if FLAGS.framework == 'pytorch' and FLAGS.torch_compile: compile_error_workloads = ['librispeech_conformer', 'ogbg', 'criteo1tb'] - eager_backend_workloads = [ - 'librispeech_deepspeech' - ] + eager_backend_workloads = ['librispeech_deepspeech'] aot_eager_backend_workloads = [] if FLAGS.workload in compile_error_workloads: logging.warning(