From a5d919cec8ddc99874b783d077bb9d273efbcf3c Mon Sep 17 00:00:00 2001 From: erogol Date: Tue, 25 Feb 2020 11:52:26 +0100 Subject: [PATCH] use 0 padding for stop tokens, which seems not working after padding_idx --- utils/data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/data.py b/utils/data.py index 87343ec1..a8b87cb5 100644 --- a/utils/data.py +++ b/utils/data.py @@ -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]]],