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

enc_out needs to be repeated at dim=1

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-28 13:44:17 +01:00
parent 438785cde1
commit 7bede84f45
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/transformer/transformer_network.py
+2 -2
View File
@@ -344,8 +344,8 @@ class TransformerPredictionNetwork(TransformerNetwork):
).unsqueeze(1)
repeated_enc_out = enc_out.repeat_interleave(
repeats=self.num_parallel_samples, dim=0
).unsqueeze(1)
repeats=self.num_parallel_samples, dim=1
)
repeated_scale = scale.repeat_interleave(
repeats=self.num_parallel_samples, dim=0