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

another typo

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-29 15:04:07 +01:00
parent 2b51a2ad25
commit d95a217801
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/transformer_tempflow/transformer_tempflow_network.py
+1 -1
View File
@@ -583,7 +583,7 @@ class TransformerTempFlowPredictionNetwork(TransformerTempFlowTrainingNetwork):
# unroll_length=1,
# )
distr_args = self.distr_args(dec_output=dec_output)
distr_args = self.distr_args(decoder_output=dec_output)
# (batch_size, 1, target_dim)
new_samples = self.flow.sample(cond=distr_args)