Open Niccolo-Ajroldi opened 2 months ago
In librispeech_conformer the model_fn returns logits_batch as a Tuple of tensors, not a tensor.
librispeech_conformer
model_fn
logits_batch
The return type is hence wrong: https://github.com/mlcommons/algorithmic-efficiency/blob/ddf5efc4e13a9a4e620ad719e9bf42303f064fac/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py#L119
It should be:
def model_fn(...) -> Tuple[Tuple[spec.Tensor, spec.Tensor], spec.ModelAuxiliaryState]:
As insignificant as it seems, this caused me quite some trouble debugging an OOM issue. Might be useful for other ppl too.
In
librispeech_conformer
themodel_fn
returnslogits_batch
as a Tuple of tensors, not a tensor.The return type is hence wrong: https://github.com/mlcommons/algorithmic-efficiency/blob/ddf5efc4e13a9a4e620ad719e9bf42303f064fac/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py#L119
It should be:
As insignificant as it seems, this caused me quite some trouble debugging an OOM issue. Might be useful for other ppl too.