debugging

mlcommons · Oct 17, 2024 · 087fd5c · 087fd5c
1 parent baac0a4
commit 087fd5c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py
@@ -113,6 +113,7 @@ def model_fn(
     variables = {'params': params, **model_state}
     inputs, input_paddings = augmented_and_preprocessed_input_batch['inputs']
     is_train_mode = mode == spec.ForwardPassMode.TRAIN
+    print(type(use_running_average_bn))
     if update_batch_norm or is_train_mode:
       (logits, logit_paddings), new_model_state = self._model.apply(
           variables,