Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "39d72bcc7b2c99c04b6f483f0d8e7bdff547d37c"
Commit a67413cc authored by Rémi Louf's avatar Rémi Louf
Browse files

extend works in-place

parent cb26b035
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
examples/run_seq2seq_finetuning.py
View file @ a67413cc
...@@ -165,7 +165,8 @@ def _fit_to_block_size(sequence, block_size): ...@@ -165,7 +165,8 @@ def _fit_to_block_size(sequence, block_size):
if len(sequence) > block_size: if len(sequence) > block_size:
return sequence[:block_size] return sequence[:block_size]
else: else:
return sequence.extend([0] * (block_size - len(sequence))) sequence.extend([0] * (block_size - len(sequence)))
return sequence
def mask_padding_tokens(sequence): def mask_padding_tokens(sequence):
......
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