Skip to content
Snippets Groups Projects
Commit d1f5ff06 authored by Ludovic Moncla's avatar Ludovic Moncla
Browse files

Update training_bertFineTuning.py

parent 5b527452
No related branches found
No related tags found
No related merge requests found
......@@ -102,7 +102,7 @@ def training_bertFineTuning(chosen_model, model_path, sentences, labels, max_le
# (3) Append the `[SEP]` token to the end.
# (4) Map tokens to their IDs.
encoded_sent = tokenizer.encode(
sent, # Sentence to encode.
str(sent), # Sentence to encode.
add_special_tokens = True, # Add '[CLS]' and '[SEP]'
# This function also supports truncation and conversion
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment