diff --git a/training_bertFineTuning.py b/training_bertFineTuning.py index bc3921e6380ebb343bf0f9c5cdffd70366609fae..dee72791e6710fa1325a43361ff3bf6110c8616a 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, )