diff --git a/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py b/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py index b5b067336..55c6e1f1b 100644 --- a/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py +++ b/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py @@ -10,6 +10,7 @@ from algorithmic_efficiency import param_utils from algorithmic_efficiency import spec +from algorithmic_efficiency.workloads.librispeech_conformer import metrics from algorithmic_efficiency.workloads.librispeech_deepspeech.librispeech_jax import \ models from algorithmic_efficiency.workloads.librispeech_deepspeech.workload import \