from local

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

View File

@ -491,6 +491,7 @@ class Data2VecAudioModel(BaseFairseqModel):
## for prompt tuning
if prompt is not None:
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: