Skip to content
Snippets Groups Projects
Commit 717e7635 authored by Ludovic Moncla's avatar Ludovic Moncla
Browse files

Update training_bertFineTuning.py

parent d1f5ff06
No related branches found
No related tags found
No related merge requests found
...@@ -199,6 +199,7 @@ def training_bertFineTuning(chosen_model, model_path, sentences, labels, max_le ...@@ -199,6 +199,7 @@ def training_bertFineTuning(chosen_model, model_path, sentences, labels, max_le
# You can increase this for multi-class tasks. # You can increase this for multi-class tasks.
output_attentions = False, # Whether the model returns attentions weights. output_attentions = False, # Whether the model returns attentions weights.
output_hidden_states = False, # Whether the model returns all hidden-states. output_hidden_states = False, # Whether the model returns all hidden-states.
local_files_only=True,
) )
elif chosen_model == 'camembert': elif chosen_model == 'camembert':
...@@ -208,6 +209,7 @@ def training_bertFineTuning(chosen_model, model_path, sentences, labels, max_le ...@@ -208,6 +209,7 @@ def training_bertFineTuning(chosen_model, model_path, sentences, labels, max_le
# You can increase this for multi-class tasks. # You can increase this for multi-class tasks.
output_attentions = False, # Whether the model returns attentions weights. output_attentions = False, # Whether the model returns attentions weights.
output_hidden_states = False, # Whether the model returns all hidden-states. output_hidden_states = False, # Whether the model returns all hidden-states.
local_files_only=True,
) )
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment