djl234 commited on
Commit
0a31ff1
1 Parent(s): b8c81e8

Update model_video.py

Browse files
Files changed (1) hide show
  1. 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 = [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)
 
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)