Update model_video.py
Browse files- model_video.py +1 -1
model_video.py
CHANGED
@@ -125,7 +125,7 @@ class Model(nn.Module):
|
|
125 |
self.concat2 = nn.ModuleList(concat_r)
|
126 |
self.concat1 = nn.ModuleList(concat_1)
|
127 |
self.mask = nn.ModuleList(mask_branch)
|
128 |
-
self.extract = [13, 23, 33, 43]
|
129 |
self.device = device
|
130 |
self.group_size = 5
|
131 |
self.intra = nn.ModuleList(intra)
|
|
|
125 |
self.concat2 = nn.ModuleList(concat_r)
|
126 |
self.concat1 = nn.ModuleList(concat_1)
|
127 |
self.mask = nn.ModuleList(mask_branch)
|
128 |
+
self.extract = [1,2,3,4,5,6,7,8,9,10,13, 23, 33, 43]
|
129 |
self.device = device
|
130 |
self.group_size = 5
|
131 |
self.intra = nn.ModuleList(intra)
|