diff --git a/training_bertFineTuning.py b/training_bertFineTuning.py index 6608e808a78dd7c39823686e65254be80aac49a4..dee72791e6710fa1325a43361ff3bf6110c8616a 100644 --- a/training_bertFineTuning.py +++ b/training_bertFineTuning.py @@ -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