needed to specify a default output_dir for the trainer

#10
Files changed (1) hide show
  1. 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...