torch compile

mlcommons · Nov 23, 2023 · 8999335 · 8999335
1 parent a77c04d
commit 8999335
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/submission_runner.py b/submission_runner.py
@@ -221,7 +221,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', 'librispeech_conformer_gelu'
+          'librispeech_conformer', 'librispeech_conformer_gelu',
           'ogbg', 'criteo1tb', 
           'imagenet_vit'
       ]