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

added missing tgt_mask

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-29 14:42:37 +01:00
parent e32ea17ce6
commit d05fc7a242
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/transformer_tempflow/transformer_tempflow_network.py
+5
View File
@@ -84,6 +84,11 @@ class TransformerTempFlowTrainingNetwork(nn.Module):
self.scaler = MeanScaler(keepdim=True)
# mask
self.register_buffer(
"tgt_mask", self.transformer.generate_square_subsequent_mask(prediction_length)
)
@staticmethod
def get_lagged_subsequences(
sequence: torch.Tensor,