get index of first pad

mattdeitke · Apr 5, 2019 · 159144c · 159144c
1 parent e2daa7c
commit 159144c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/babyai/model.py b/babyai/model.py
@@ -256,6 +256,8 @@ def forward(self, obs, memory, instr_embedding=None):
     def _get_instr_embedding(self, instr):
         if self.lang_model == 'gru':
             _, hidden = self.instr_rnn(self.word_embedding(instr))
+            x = (instr <= 0).sum(dim=1)
+            print(x)
             return hidden[-1]
 
         elif self.lang_model in ['bigru', 'attgru']: