fix

mlcommons · Oct 2, 2023 · e71a530 · e71a530
1 parent 4e08106
commit e71a530
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/models.py b/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/models.py
@@ -372,7 +372,8 @@ def __call__(
 
     if use_cuda:
       logging.info('lstm input')
-      logging.info(inputs)
+      jax.debug.print(inputs)
+      # logging.info(inputs)
       y, h, c = rnn.lstm(
           x=inputs, h_0=h_0, c_0=c_0, weights=weights,
           seq_lengths=seq_lengths, input_size=input_size,