From 6e9f7af37be3d9fcd133b0c07ca3fd9eb05f62de Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Wed, 30 Mar 2022 21:39:28 +0200 Subject: [PATCH] typo --- tft/module.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/tft/module.py b/tft/module.py index 0140fd4..55afb67 100644 --- a/tft/module.py +++ b/tft/module.py @@ -569,9 +569,7 @@ class TFTModel(nn.Module): c_c = self.state_c(static_var) states = [c_h.unsqueeze(0), c_c.unsqueeze(0)] - enc_out = self.temporal_encoder( - past_selection, tgt_input=None, states=states - ) + enc_out = self.temporal_encoder(past_selection, tgt_input=None, states=states) dec_output = self.temporal_decoder(enc_out, static_enrichment) params = self.param_proj(dec_output) @@ -580,12 +578,13 @@ class TFTModel(nn.Module): next_sample = distr.sample() future_samples = [next_sample] import pdb + pdb.set_trace() for k in range(1, self.prediction_length): # TODO - future_target_proj = pass - future_time_feat_proj = pass + # future_target_proj = pass + # future_time_feat_proj = pass future_selection, _ = self.future_selection( [future_target_proj, future_time_feat_proj], static_selection