You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
def forward(self,features,attentions):
features = list(features)
if isinstance(attentions,dict):
for i in range(len(attentions)):
if i==0:
atten = attentions[i].unsqueeze(0)
else:
atten = torch.cat((atten,attentions[i].unsqueeze(0)),dim=0)
attentions = atten.cuda()
output = []
h , w = attentions.shape[2:]
ncls = attentions.shape[0]
key_t = self.key_t(attentions)
val_t = self.value_t(attentions)
for idx in range(len(features)):
我想知道谁是support ,谁是query?按代码的意思,是attentions是support,features是query是这样吗?
attentions和features的维度分别是多少?
期待您的回复,谢谢!
The text was updated successfully, but these errors were encountered: