Skip to content

GitLab

Commit 364920e2 authored by VictorSanh's avatar VictorSanh
Browse files

fix small bug/typo

parent 23c23f53
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/distillation/dataset.py
View file @ 364920e2
...@@ -77,7 +77,7 @@ class Dataset: ...@@ -77,7 +77,7 @@ class Dataset:
if sub_s[0] != cls_id: if sub_s[0] != cls_id:
sub_s = np.insert(sub_s, 0, cls_id) sub_s = np.insert(sub_s, 0, cls_id)
if sub_s[-1] != sep_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 assert len(sub_s) <= max_len
sub_seqs.append(sub_s) sub_seqs.append(sub_s)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment