6db2a40542917f11509cbdc79c882049931d4f86,src/agent.py,BatchedRolloutAgent,write,#BatchedRolloutAgent#,255

Before Change

                lang_h, sent_lang_hs, move = states[sent]
                self.lang_h = lang_h
                self.lang_hs.append(sent_lang_hs)
                self.words.append(self.model.word2var("YOU:"))
                self.words.append(self.model.to_device(Variable(move)))
                assert (torch.cat(self.words).size()[0] == torch.cat(self.lang_hs).size()[0])

After Change

                lang_h, sent_lang_hs, move = states[sent]
                self.lang_h = lang_h
                self.lang_hs.append(sent_lang_hs)
                self.words.append(self.model.word2var("YOU:").unsqueeze(1))
                self.words.append(self.model.to_device(Variable(move)))
                assert (torch.cat(self.words).size()[0] == torch.cat(self.lang_hs).size()[0])

Instances