from local

This commit is contained in:
dohe0342 2023-04-26 12:17:37 +09:00
parent 68be92565c
commit 2d4961e6ff
2 changed files with 1 additions and 0 deletions

View File

@ -494,6 +494,7 @@ class Data2VecAudioModel(BaseFairseqModel):
for i in range(padding_mask.size()[0]):
nonzero = padding_mask[i].nonzero()
try:
print(nonzero[0])
conv_feat_all = torch.cat([conv_feat_all, features[i][nonzero[0]]])
except:
conv_feat_all = torch.cat([conv_feat_all, features])