6db2a40542917f11509cbdc79c882049931d4f86,src/agent.py,LstmRolloutAgent,write,#LstmRolloutAgent#,191

Before Change

        outs, lang_h, lang_hs = res
        self.lang_h = lang_h
        self.lang_hs.append(lang_hs)
        self.words.append(self.model.word2var("YOU:"))
        self.words.append(outs)
        return self._decode(outs, self.model.word_dict)

After Change

        outs, lang_h, lang_hs = res
        self.lang_h = lang_h
        self.lang_hs.append(lang_hs)
        self.words.append(self.model.word2var("YOU:").unsqueeze(1))
        self.words.append(outs)
        return self._decode(outs, self.model.word_dict)

Instances