from local

This commit is contained in:
dohe0342 2023-04-26 12:19:25 +09:00
parent 9d80f13024
commit 9a422296d7
2 changed files with 1 additions and 1 deletions

View File

@ -497,7 +497,7 @@ class Data2VecAudioModel(BaseFairseqModel):
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])
conv_feat_all = torch.cat([conv_feat_all, features[i]])
print(conv_feat_all.size())
exit()
prompt = prompt.expand((features.size()[0], prompt.size()[0], prompt.size()[1]))