wzkariampuzha commited on
Commit
77f23c9
1 Parent(s): e85b9c4

Update epi_pipeline.py

Browse files
Files changed (1) hide show
  1. 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: ./tmp_trainer,
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...