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

permute back the transformer output for conditioning

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-29 15:14:25 +01:00
parent d95a217801
commit 721adece9f
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(decoder_output=dec_output)
distr_args = self.distr_args(decoder_output=dec_output.permute(1,0,2))
# (batch_size, 1, target_dim)
new_samples = self.flow.sample(cond=distr_args)