From 8d99197bb479aa2cccadadd4f0def63d5c39c176 Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Wed, 8 Nov 2023 01:18:08 +0000 Subject: [PATCH] formatting --- submission_runner.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/submission_runner.py b/submission_runner.py index b54513a50..7c7b1371f 100644 --- a/submission_runner.py +++ b/submission_runner.py @@ -220,7 +220,9 @@ 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 = ['librispeech_conformer', 'ogbg', 'criteo1tb', 'imagenet_vit'] + compile_error_workloads = [ + 'librispeech_conformer', 'ogbg', 'criteo1tb', 'imagenet_vit' + ] eager_backend_workloads = ['librispeech_deepspeech'] aot_eager_backend_workloads = [] if FLAGS.workload in compile_error_workloads: