nanushio commited on
Commit
a877246
1 Parent(s): c0d774a

- [MINOR] [SOURCE] [UPDATE] 1. update app.py

Browse files
Files changed (1) hide show
  1. cover/models/evaluator.py +1 -1
cover/models/evaluator.py CHANGED
@@ -142,7 +142,7 @@ class COVER(nn.Module):
142
  else:
143
  scores += [getattr(self, "vqa_head")(feat_gated)]
144
  elif key == 'semantic':
145
- x = vclips[key]#.squeeze()
146
  x = x.permute(1,0,2,3)
147
  feat, _ = getattr(self, key.split("_")[0] + "_backbone")(
148
  x, multi=self.multi, layer=self.layer, **kwargs
 
142
  else:
143
  scores += [getattr(self, "vqa_head")(feat_gated)]
144
  elif key == 'semantic':
145
+ x = vclips[key].squeeze(0)
146
  x = x.permute(1,0,2,3)
147
  feat, _ = getattr(self, key.split("_")[0] + "_backbone")(
148
  x, multi=self.multi, layer=self.layer, **kwargs