Data loader bug fix 2

2026-09-09 11:16:00 +08:00 · 2018-03-26 11:07:15 -07:00
parent 3c084177c6
commit 1ff8d6d2b7
4 changed files with 9 additions and 5 deletions
@@ -256,9 +256,12 @@ class Decoder(nn.Module):
        greedy = not self.training

        if memory is not None:
+            print(memory.shape)
+            
            # Grouping multiple frames if necessary
            if memory.size(-1) == self.memory_dim:
                memory = memory.view(B, memory.size(1) // self.r, -1)
+            print(memory.shape)
            assert memory.size(-1) == self.memory_dim * self.r,\
                " !! Dimension mismatch {} vs {} * {}".format(memory.size(-1),
                                                         self.memory_dim, self.r)