Spaces:
Sleeping
Sleeping
wzkariampuzha
commited on
Commit
•
77f23c9
1
Parent(s):
e85b9c4
Update epi_pipeline.py
Browse files- epi_pipeline.py +1 -1
epi_pipeline.py
CHANGED
@@ -657,7 +657,7 @@ class NER_Pipeline:
|
|
657 |
self.labels = {re.sub(".-","",label) for label in self.config.label2id.keys() if label != "O"}
|
658 |
# Create the TrainingArguments object and set the batch size
|
659 |
training_args = TrainingArguments(
|
660 |
-
output_dir
|
661 |
per_device_eval_batch_size=16,
|
662 |
per_gpu_eval_batch_size=16,
|
663 |
# other training arguments...
|
|
|
657 |
self.labels = {re.sub(".-","",label) for label in self.config.label2id.keys() if label != "O"}
|
658 |
# Create the TrainingArguments object and set the batch size
|
659 |
training_args = TrainingArguments(
|
660 |
+
output_dir="./tmp_trainer",
|
661 |
per_device_eval_batch_size=16,
|
662 |
per_gpu_eval_batch_size=16,
|
663 |
# other training arguments...
|