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

make mask optional for now

Browse Source
This commit is contained in:
Kashif Rasul
2022-03-30 17:04:29 +02:00
parent 8c8b1aa4eb
commit 248f81e6c2
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
@@ -219,7 +219,7 @@ class TemporalFusionDecoder(nn.Module):
return mask
def forward(
self, x: torch.Tensor, static: torch.Tensor, mask: torch.Tensor
self, x: torch.Tensor, static: torch.Tensor, mask: Optional[torch.Tensor] = None
) -> torch.Tensor:
static = static.repeat((1, self.context_length + self.prediction_length, 1))