[UPD] mask and resources
Browse files
model.py
CHANGED
|
@@ -86,7 +86,7 @@ class ZeroSwotEncoderModel(PreTrainedModel):
|
|
| 86 |
# BOS and EOS embeddings
|
| 87 |
x, mask = self.speech_embedder(x, mask) # [B, N+2, D]
|
| 88 |
|
| 89 |
-
return x, mask
|
| 90 |
|
| 91 |
|
| 92 |
class SpeechEmbedder(nn.Module):
|
|
|
|
| 86 |
# BOS and EOS embeddings
|
| 87 |
x, mask = self.speech_embedder(x, mask) # [B, N+2, D]
|
| 88 |
|
| 89 |
+
return x, ~mask
|
| 90 |
|
| 91 |
|
| 92 |
class SpeechEmbedder(nn.Module):
|
methodology.png → resources/methodology.png
RENAMED
|
File without changes
|
sample.wav → resources/sample.wav
RENAMED
|
File without changes
|