From 717e7635e26f0b0700905628098cdfb7068e7361 Mon Sep 17 00:00:00 2001
From: Ludovic Moncla <ludovic.moncla@insa-lyon.fr>
Date: Mon, 20 Sep 2021 12:35:45 +0000
Subject: [PATCH] Update training_bertFineTuning.py
---
training_bertFineTuning.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/training_bertFineTuning.py b/training_bertFineTuning.py
index dee7279..bc3921e 100644
--- a/training_bertFineTuning.py
+++ b/training_bertFineTuning.py
@@ -199,6 +199,7 @@ def training_bertFineTuning(chosen_model, model_path, sentences, labels, max_le
# You can increase this for multi-class tasks.
output_attentions = False, # Whether the model returns attentions weights.
output_hidden_states = False, # Whether the model returns all hidden-states.
+ local_files_only=True,
)
elif chosen_model == 'camembert':
@@ -208,6 +209,7 @@ def training_bertFineTuning(chosen_model, model_path, sentences, labels, max_le
# You can increase this for multi-class tasks.
output_attentions = False, # Whether the model returns attentions weights.
output_hidden_states = False, # Whether the model returns all hidden-states.
+ local_files_only=True,
)
--
GitLab