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