c-my/transformers
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix small bug/typo

Browse Source
This commit is contained in:
VictorSanh committed 2019-09-10 21:45:01 +00:00
1 parent 23c23f5399
commit 364920e216
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
examples/distillation/dataset.py
+1 -1
View File
@@ -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)