from local

This commit is contained in:
dohe0342 2023-04-11 13:49:58 +09:00
parent 37281337b5
commit b0554208fc
3 changed files with 3 additions and 0 deletions

View File

@ -494,6 +494,9 @@ class Data2VecAudioModel(BaseFairseqModel):
else:
padding_mask = None
print(orig_padding_mask.size())
print(padding_mask.size())
if self.post_extract_proj is not None:
features = self.post_extract_proj(features)