From 729c4a1b575fd3ec54cef68f3b2f7b0eee738274 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Fri, 11 Jun 2021 12:51:44 +0200 Subject: [PATCH] fix the scaler --- pts/model/causal_deepar/causal_deepar_network.py | 4 ++-- pts/model/deepar/deepar_network.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/pts/model/causal_deepar/causal_deepar_network.py b/pts/model/causal_deepar/causal_deepar_network.py index 64a9c2c..ff1e787 100644 --- a/pts/model/causal_deepar/causal_deepar_network.py +++ b/pts/model/causal_deepar/causal_deepar_network.py @@ -177,8 +177,8 @@ class CausalDeepARNetwork(nn.Module): # scale is computed on the context length last units of the past target # scale shape is (batch_size, 1, *target_shape) _, scale = self.scaler( - past_target[:, self.context_length :, ...], - past_observed_values[:, self.context_length :, ...], + past_target[:, -self.context_length :, ...], + past_observed_values[:, -self.context_length :, ...], ) _, control_scale = self.control_scaler( diff --git a/pts/model/deepar/deepar_network.py b/pts/model/deepar/deepar_network.py index bd98e05..12f1709 100644 --- a/pts/model/deepar/deepar_network.py +++ b/pts/model/deepar/deepar_network.py @@ -160,8 +160,8 @@ class DeepARNetwork(nn.Module): # scale is computed on the context length last units of the past target # scale shape is (batch_size, 1, *target_shape) _, scale = self.scaler( - past_target[:, self.context_length :, ...], - past_observed_values[:, self.context_length :, ...], + past_target[:, -self.context_length :, ...], + past_observed_values[:, -self.context_length :, ...], ) # (batch_size, num_features)