From 86b3a6de04166cbfc9082822ef620f2d45aff6db Mon Sep 17 00:00:00 2001 From: Suraj Subramanian <5676233+suraj813@users.noreply.github.com> Date: Fri, 7 Oct 2022 14:20:51 -0400 Subject: [PATCH] Update transformer_tutorial.py --- 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 95372cabceb..e3198e855a2 100644 --- a/beginner_source/transformer_tutorial.py +++ b/beginner_source/transformer_tutorial.py @@ -202,7 +202,7 @@ def batchify(data: Tensor, bsz: int) -> Tensor: """ seq_len = data.size(0) // bsz data = data[:seq_len * bsz] - data = data.view(bsz, seq_len).t().contiguous() + data = data.view(bsz, seq_len).contiguous() return data.to(device) batch_size = 20