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

fix typo

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-27 17:00:09 +01:00
parent 3184f5d588
commit 4a1de2d06f
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/transformer/transformer_network.py
+3 -3
View File
@@ -33,7 +33,7 @@ class TransformerNetwork(nn.Module):
) -> None:
super().__init__(**kwargs)
self.history_length = history_length
self.history_length = history_length
self.context_length = context_length
self.prediction_length = prediction_length
self.scaling = scaling
@@ -252,7 +252,7 @@ class TransformerTrainingNetwork(TransformerNetwork):
dec_output = self.decoder(
dec_input,
enc_out,
self.upper_triangular_mask(F, self.prediction_length),
self.upper_triangular_mask(self.prediction_length),
)
# compute loss
@@ -352,7 +352,7 @@ class TransformerPredictionNetwork(TransformerNetwork):
dim=-1,
)
dec_output = self.decoder(dec_input, repeated_enc_out, None, False)
dec_output = self.decoder(dec_input, repeated_enc_out, None, False) # TODO check False argument
distr_args = self.proj_dist_args(dec_output)