From 364920e216c16d73c782a61a4cf6652e541fbe18 Mon Sep 17 00:00:00 2001 From: VictorSanh Date: Tue, 10 Sep 2019 21:45:01 +0000 Subject: [PATCH] fix small bug/typo --- examples/distillation/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/distillation/dataset.py b/examples/distillation/dataset.py index cdc16b94f3..89e3f1187f 100644 --- a/examples/distillation/dataset.py +++ b/examples/distillation/dataset.py @@ -77,7 +77,7 @@ class Dataset: if sub_s[0] != cls_id: sub_s = np.insert(sub_s, 0, cls_id) if sub_s[-1] != sep_id: - sub_s = np.insert(sub_s, len(sub_s), cls_id) + sub_s = np.insert(sub_s, len(sub_s), sep_id) assert len(sub_s) <= max_len sub_seqs.append(sub_s)