Merge branch 'develop' into baker

2021-03-29 11:17:51 +08:00 · 2021-03-29 11:17:51 +08:00 · 4757f08550
parent ab85d5ca13 59ed247840
commit 4757f08550
1 changed files with 1 additions and 1 deletions
--- a/parakeet/models/lstm_speaker_encoder.py
+++ b/parakeet/models/lstm_speaker_encoder.py
@ -33,7 +33,7 @@ class LSTMSpeakerEncoder(nn.Layer):
        if reduce:
            embed = paddle.mean(normalized_embeds, 0)
            embed = F.normalize(embed, axis=0)            
-        return embeds
+        return embed
    
    def embed_utterance(self, utterances, initial_states=None):
        # utterances: [B, T, C] -> embed [C']