from local

This commit is contained in:
dohe0342 2023-04-26 12:20:09 +09:00
parent 8284131203
commit 9b05cef7e8
2 changed files with 1 additions and 1 deletions

View File

@ -490,8 +490,8 @@ class Data2VecAudioModel(BaseFairseqModel):
## for prompt tuning
if prompt is not None:
print(features.size())
conv_feat_all = torch.tensor([]).to(features.device)
print(conv_feat_all.size())
for i in range(padding_mask.size()[0]):
nonzero = padding_mask[i].nonzero()
try: