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

fix typo

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-29 14:51:36 +01:00
parent 054104019c
commit 50893b3f13
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
@@ -546,7 +546,7 @@ class TransformerTempFlowPredictionNetwork(TransformerTempFlowTrainingNetwork):
subsequences_length=1,
)
lags_scaled = lags / repeated_scale.unsqueeze(1)
lags_scaled = lags / repeated_scale.unsqueeze(-1)
input_lags = lags_scaled.reshape(
shape=(-1, 1, prod(self.target_shape) * len(self.lags_seq))