Merge pull request shentianxiao#11 from M3skar/master

fix model.py indent error
trancongman276 · Jun 28, 2021 · 3a32f43 · 3a32f43
2 parents a096cbe + 37e9b64
commit 3a32f43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -67,7 +67,7 @@ def decode(self, z, input, hidden=None):
         logits = self.proj(output.view(-1, output.size(-1)))
         return logits.view(output.size(0), output.size(1), -1), hidden
 
-     def generate(self, z, max_len, alg):
+    def generate(self, z, max_len, alg):
         assert alg in ['greedy' , 'sample' , 'top5']
         sents = []
         input = torch.zeros(1, len(z), dtype=torch.long, device=z.device).fill_(self.vocab.go)