From 623b69a21984cd93689b8dfc937c194ddf35f072 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Wed, 30 Mar 2022 22:25:19 +0200 Subject: [PATCH] typo --- tft/module.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tft/module.py b/tft/module.py index 7420b7c..cdf13d8 100644 --- a/tft/module.py +++ b/tft/module.py @@ -586,6 +586,9 @@ class TFTModel(nn.Module): repeated_past_selection = past_selection.repeat_interleave( repeats=self.num_parallel_samples, dim=0 ) + repeated_static_selection = static_selection.repeat_interleave( + repeats=self.num_parallel_samples, dim=0 + ) repeated_static_enrichment = static_enrichment.repeat_interleave( repeats=self.num_parallel_samples, dim=0 ) @@ -609,7 +612,8 @@ class TFTModel(nn.Module): next_time_feat_proj = repeated_time_feat_proj[:, k : k + 1] future_selection, _ = self.future_selection( - [reshaped_lagged_sequence_proj, next_time_feat_proj], static_selection + [reshaped_lagged_sequence_proj, next_time_feat_proj], + repeated_static_selection, ) enc_out = self.temporal_encoder( repeated_past_selection, future_selection, repeated_states