nanushio
commited on
Commit
•
a877246
1
Parent(s):
c0d774a
- [MINOR] [SOURCE] [UPDATE] 1. update app.py
Browse files
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]
|
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
|