from local
This commit is contained in:
parent
f41a2ee2a1
commit
114cc475a7
Binary file not shown.
@ -494,7 +494,7 @@ class Data2VecAudioModel(BaseFairseqModel):
|
|||||||
else:
|
else:
|
||||||
padding_mask = None
|
padding_mask = None
|
||||||
|
|
||||||
print(padding_mask[9].index(True))
|
print((padding_mask[9] == True).nonzero(as_tuple=True)[0])
|
||||||
exit()
|
exit()
|
||||||
|
|
||||||
if self.post_extract_proj is not None:
|
if self.post_extract_proj is not None:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user