From d1f5ff06e0b69244e9d3d3733280f5237cb87163 Mon Sep 17 00:00:00 2001
From: Ludovic Moncla <ludovic.moncla@insa-lyon.fr>
Date: Mon, 20 Sep 2021 12:06:41 +0000
Subject: [PATCH] Update training_bertFineTuning.py
---
training_bertFineTuning.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/training_bertFineTuning.py b/training_bertFineTuning.py
index 6608e80..dee7279 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
--
GitLab