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

mask with diagonal=1

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-28 14:29:06 +01:00
parent ad482cb526
commit fcb69e83d2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/transformer/transformer_network.py
+1 -1
View File
@@ -83,7 +83,7 @@ class TransformerNetwork(nn.Module):
# mask
self.register_buffer(
"tgt_mask", torch.triu(torch.ones((prediction_length, prediction_length)))
"tgt_mask", torch.triu(torch.ones((prediction_length, prediction_length)), diagonal=1)
)
@staticmethod