wassname/pytorch-transformer-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-transformer-ts.git synced 2026-08-12 12:20:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

typo

Browse Source
This commit is contained in:
Kashif Rasul
2022-03-30 21:39:28 +02:00
parent 8efc4a7b63
commit 6e9f7af37b
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tft/module.py
+4 -5
View File
@@ -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