wassname/pytorch-transformer-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-transformer-ts.git synced 2026-09-12 12:40:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

query length

Browse Source
This commit is contained in:
Kashif Rasul
2022-03-30 22:51:36 +02:00
parent 623b69a219
commit 2918cbd13e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tft/module.py
+1 -1
View File
@@ -242,7 +242,7 @@ class TemporalFusionDecoder(nn.Module):
query_key_value = x
attn_output, _ = self.attention(
query=query_key_value[:, -self.prediction_length :, ...],
query=query_key_value[:, self.context_length :, ...],
key=query_key_value,
value=query_key_value,
# key_padding_mask=key_padding_mask,