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
# 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,
)
......
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