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

fix repeated_past_target

Browse Source
This commit is contained in:
Kashif Rasul
2022-03-31 12:07:14 +02:00
parent e198ba4544
commit 33cf2402a6
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tft/module.py
+3 -2
View File
@@ -583,8 +583,9 @@ class TFTModel(nn.Module):
repeats=self.num_parallel_samples, dim=0
)
repeated_past_target = target.repeat_interleave(
repeats=self.num_parallel_samples, dim=0
repeated_past_target = (
past_target.repeat_interleave(repeats=self.num_parallel_samples, dim=0)
/ repeated_scale
)
repeated_past_selection = past_selection.repeat_interleave(
repeats=self.num_parallel_samples, dim=0