Merge branch 'master' of github.com:gauenk/stnls

gauenk · Dec 2, 2023 · 14f2784 · 14f2784
2 parents 0234029 + fb5e704
commit 14f2784
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lib/stnls/agg/stack_conv.py b/lib/stnls/agg/stack_conv.py
@@ -32,6 +32,7 @@ def forward(self,vid,weights,flows):
         stack = self.stacker(vid,weights,flows)
         stack = rearrange(stack,'b hd k t c h w -> b t k (hd c) h w')
         vid = self.run_projection(stack)
+        # print("[stack_conv] vid.shape: ",vid.shape)
         return vid
 
     def run_projection(self,stack):