wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-07-17 11:32:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

formatting

Browse Source
This commit is contained in:
Kashif Rasul
2020-01-27 20:05:12 +01:00
parent da584d309c
commit eea4d90aba
1 changed files with 17 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/transformer/transformer_network.py
+17 -17
View File
@@ -140,24 +140,24 @@ class TransformerNetwork(nn.Module):
# axis=1,
# begin=self.history_length - self.context_length,
# end=None,
# )
sequence = past_target
sequence_length = self.history_length
subsequences_length = self.context_length
else:
time_feat = torch.cat((
past_time_feat[:, self.history_length -
self.context_length:, ...], # .slice_axis(
# axis=1,
# begin=self.history_length - self.context_length,
# end=None,
# ),
future_time_feat),
dim=1,
)
sequence=past_target
sequence_length=self.history_length
subsequences_length=self.context_length
else:
time_feat=torch.cat((
past_time_feat[:, self.history_length -
self.context_length:, ...], # .slice_axis(
# axis=1,
# begin=self.history_length - self.context_length,
# end=None,
# ),
future_time_feat),
dim=1,
)
sequence = torch.cat((past_target, future_target), dim=1)
sequence_length = self.history_length + self.prediction_length
subsequences_length = self.context_length + self.prediction_length
sequence = torch.cat((past_target, future_target), dim=1)
sequence_length = self.history_length + self.prediction_length
subsequences_length = self.context_length + self.prediction_length
# (batch_size, sub_seq_len, *target_shape, num_lags)
lags = self.get_lagged_subsequences(