Fix setting tgt when using a dataset with a performance less than the…

… max sequence length
gwinndr · May 26, 2022 · 7161590 · 7161590
1 parent 4ea0833
commit 7161590
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dataset/e_piano.py b/dataset/e_piano.py
@@ -89,7 +89,7 @@ def process_midi(raw_mid, max_seq, random_seq):
     if(raw_len < full_seq):
         x[:raw_len]         = raw_mid
         tgt[:raw_len-1]     = raw_mid[1:]
-        tgt[raw_len]        = TOKEN_END
+        tgt[raw_len-1]      = TOKEN_END
     else:
         # Randomly selecting a range
         if(random_seq):