Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

fix SpeechRecognitionOutputTransform if local #1362

Merged
merged 8 commits into from
Jun 29, 2022
4 changes: 3 additions & 1 deletion flash/audio/speech_recognition/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,9 @@ def __init__(
optimizer=optimizer,
lr_scheduler=lr_scheduler,
learning_rate=learning_rate,
output_transform=SpeechRecognitionOutputTransform(backbone),
output_transform=SpeechRecognitionOutputTransform(backbone)
if processor_backbone is None
else SpeechRecognitionOutputTransform(processor_backbone),
)

self.save_hyperparameters()
Expand Down