Spaces:
Sleeping
Sleeping
wzkariampuzha
commited on
Commit
•
e85b9c4
1
Parent(s):
3a18b1f
needed to specify a default output_dir for the trainer (#10)
Browse files- needed to specify a default output_dir for the trainer (62db55402785525f6213a03c345e5b3828939965)
- epi_pipeline.py +1 -0
epi_pipeline.py
CHANGED
@@ -657,6 +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 |
per_device_eval_batch_size=16,
|
661 |
per_gpu_eval_batch_size=16,
|
662 |
# 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...
|