From c443f187b432ed865e139962900d1a1e55629d96 Mon Sep 17 00:00:00 2001 From: Ludovic Moncla <ludovic.moncla@insa-lyon.fr> Date: Mon, 20 Sep 2021 13:05:36 +0000 Subject: [PATCH] Update training_bertFineTuning.py --- training_bertFineTuning.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/training_bertFineTuning.py b/training_bertFineTuning.py index bc3921e..dee7279 100644 --- a/training_bertFineTuning.py +++ b/training_bertFineTuning.py @@ -199,7 +199,6 @@ 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': @@ -209,7 +208,6 @@ 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