fix comment

kashif · Oct 17, 2023 · 7f55e5e · 7f55e5e
1 parent 5ea1e2d
commit 7f55e5e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gluonts/torch/model/i_transformer/module.py b/src/gluonts/torch/model/i_transformer/module.py
@@ -172,7 +172,7 @@ def forward(
             self.d_model // self.nhead,
         )
 
-        # transpose to variate first
+        # transpose to prediction length first
         projection_out_transpose = projection_out.transpose(1, 2)
 
         # project to distribution arguments