From 54d568250098cdafbb475f0fd03fe7dc6e86363f Mon Sep 17 00:00:00 2001 From: ziang Date: Fri, 11 Jun 2021 22:51:00 +0800 Subject: [PATCH] change torch.div to torch.floor_divide @line81 --- fairseq/search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fairseq/search.py b/fairseq/search.py index afcc388..82896b7 100755 --- a/fairseq/search.py +++ b/fairseq/search.py @@ -78,7 +78,7 @@ def step(self, step, lprobs, scores): ), out=(self.scores_buf, self.indices_buf), ) - torch.div(self.indices_buf, vocab_size, out=self.beams_buf) + torch.floor_divide(self.indices_buf, vocab_size, out=self.beams_buf) self.indices_buf.fmod_(vocab_size) return self.scores_buf, self.indices_buf, self.beams_buf