Merge branch 'develop' into baker
This commit is contained in:
commit
4757f08550
|
@ -33,7 +33,7 @@ class LSTMSpeakerEncoder(nn.Layer):
|
||||||
if reduce:
|
if reduce:
|
||||||
embed = paddle.mean(normalized_embeds, 0)
|
embed = paddle.mean(normalized_embeds, 0)
|
||||||
embed = F.normalize(embed, axis=0)
|
embed = F.normalize(embed, axis=0)
|
||||||
return embeds
|
return embed
|
||||||
|
|
||||||
def embed_utterance(self, utterances, initial_states=None):
|
def embed_utterance(self, utterances, initial_states=None):
|
||||||
# utterances: [B, T, C] -> embed [C']
|
# utterances: [B, T, C] -> embed [C']
|
||||||
|
|
Loading…
Reference in New Issue