From 50893b3f133f20975de4564fbd9ec3a96b12efcf Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Wed, 29 Jan 2020 14:51:36 +0100 Subject: [PATCH] fix typo --- pts/model/transformer_tempflow/transformer_tempflow_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pts/model/transformer_tempflow/transformer_tempflow_network.py b/pts/model/transformer_tempflow/transformer_tempflow_network.py index f01cdba..69ac2b9 100644 --- a/pts/model/transformer_tempflow/transformer_tempflow_network.py +++ b/pts/model/transformer_tempflow/transformer_tempflow_network.py @@ -546,7 +546,7 @@ class TransformerTempFlowPredictionNetwork(TransformerTempFlowTrainingNetwork): subsequences_length=1, ) - lags_scaled = lags / repeated_scale.unsqueeze(1) + lags_scaled = lags / repeated_scale.unsqueeze(-1) input_lags = lags_scaled.reshape( shape=(-1, 1, prod(self.target_shape) * len(self.lags_seq))