From 7fe6485cddd1d46b3bfc373a90b7b0018e070bd2 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Wed, 30 Mar 2022 23:25:15 +0200 Subject: [PATCH] use scaled target --- tft/module.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tft/module.py b/tft/module.py index 119c824..96c3c34 100644 --- a/tft/module.py +++ b/tft/module.py @@ -579,9 +579,8 @@ class TFTModel(nn.Module): repeats=self.num_parallel_samples, dim=0 ) - repeated_past_target = ( - past_target.repeat_interleave(repeats=self.num_parallel_samples, dim=0) - / repeated_scale + repeated_past_target = target.repeat_interleave( + repeats=self.num_parallel_samples, dim=0 ) repeated_past_selection = past_selection.repeat_interleave( repeats=self.num_parallel_samples, dim=0