from local

This commit is contained in:
dohe0342 2023-04-26 12:16:07 +09:00
parent 6d686fe729
commit ac3b3d594c
2 changed files with 1 additions and 1 deletions

View File

@ -493,7 +493,7 @@ class Data2VecAudioModel(BaseFairseqModel):
print(features.size())
print(padding_mask.size())
print(padding_mask[0])
conv_feat_all = torch.tensor([])
conv_feat_all = torch.tensor([]).to(features.device())
for i in range(padding_mask.size()[0]):
nonzero = padding_mask[i].nonzero()
try: