use 0 padding for stop tokens, which seems not working after padding_idx

2026-09-09 11:16:00 +08:00 · 2020-02-25 11:52:26 +01:00
parent e6504cc9a4
commit a5d919cec8
1 changed files with 1 additions and 1 deletions
@@ -14,7 +14,7 @@ def prepare_data(inputs):


 def _pad_tensor(x, length):
-    _pad = 0
+    _pad = 0.
    assert x.ndim == 2
    x = np.pad(
        x, [[0, 0], [0, length - x.shape[1]]],