typo

mlcommons · Sep 26, 2023 · 28db392 · 28db392
1 parent 6b9655f
commit 28db392
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py b/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py
@@ -40,7 +40,7 @@ def init_model_fn(
                               console_kwargs={'force_terminal': False,
                                               'force_jupyter': False,
                                               'width': 240})
-    logging.info(tabulate_fn(*fake_input_batch), train=False)
+    logging.info(tabulate_fn(*fake_input_batch, train=False))
     model_init_fn = jax.jit(functools.partial(self._model.init, train=False))
 
     params_rng, dropout_rng = jax.random.split(rng, 2)