From 6b9655f9edc96d421b971f15febe0eaeba7a57a2 Mon Sep 17 00:00:00 2001 From: priyakasimbeg Date: Tue, 26 Sep 2023 23:07:13 +0000 Subject: [PATCH] tabulate typo fix --- .../librispeech_deepspeech/librispeech_jax/workload.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py b/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py index 54da37f0e..46c524381 100644 --- 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(tabuleate_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)