From 2b51a2ad2540fd528923c2e55d15f5f09aed0120 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Wed, 29 Jan 2020 15:00:46 +0100 Subject: [PATCH] use target dim to reshape lags --- 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 69ac2b9..cd82f10 100644 --- a/pts/model/transformer_tempflow/transformer_tempflow_network.py +++ b/pts/model/transformer_tempflow/transformer_tempflow_network.py @@ -549,7 +549,7 @@ class TransformerTempFlowPredictionNetwork(TransformerTempFlowTrainingNetwork): lags_scaled = lags / repeated_scale.unsqueeze(-1) input_lags = lags_scaled.reshape( - shape=(-1, 1, prod(self.target_shape) * len(self.lags_seq)) + (-1, 1, len(self.lags_seq) * self.target_dim) ) # (batch_size, target_dim, embed_dim)