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

use target dim to reshape lags

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-29 15:00:46 +01:00
parent 50893b3f13
commit 2b51a2ad25
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/transformer_tempflow/transformer_tempflow_network.py
+1 -1
View File
@@ -549,7 +549,7 @@ class TransformerTempFlowPredictionNetwork(TransformerTempFlowTrainingNetwork):
lags_scaled = lags / repeated_scale.unsqueeze(-1)
input_lags = lags_scaled.reshape(
shape=(-1, 1, prod(self.target_shape) * len(self.lags_seq))
(-1, 1, len(self.lags_seq) * self.target_dim)
)
# (batch_size, target_dim, embed_dim)