From 5989e5592c60a2e6a72caa9468aa616af0126e92 Mon Sep 17 00:00:00 2001 From: David Pollack Date: Fri, 11 Oct 2019 14:32:56 -0700 Subject: [PATCH] Fix CrossEntropyLoss in transformers tutorial Signed-off-by: David Pollack --- beginner_source/transformer_tutorial.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/beginner_source/transformer_tutorial.py b/beginner_source/transformer_tutorial.py index 8ee29ce7b4e..0ba9711ed67 100644 --- a/beginner_source/transformer_tutorial.py +++ b/beginner_source/transformer_tutorial.py @@ -86,7 +86,7 @@ def forward(self, src): src = self.pos_encoder(src) output = self.transformer_encoder(src, self.src_mask) output = self.decoder(output) - return F.log_softmax(output, dim=-1) + return output ######################################################################