Spaces:

ncats
/

EpiPipeline4RD

Running

wzkariampuzha commited on May 31, 2023

Commit

0605915

•

1 Parent(s): 77f23c9

Update epi_pipeline.py

Files changed (1) hide show

epi_pipeline.py CHANGED Viewed

@@ -659,7 +659,6 @@ class NER_Pipeline:
         training_args = TrainingArguments(
             output_dir="./tmp_trainer",
             per_device_eval_batch_size=16,
-            per_gpu_eval_batch_size=16,
             # other training arguments...
             )
         self.trainer = Trainer(model=AutoModelForTokenClassification.from_pretrained(name_or_path_to_model_folder),args=training_args)

         training_args = TrainingArguments(
             output_dir="./tmp_trainer",
             per_device_eval_batch_size=16,
             # other training arguments...
             )
         self.trainer = Trainer(model=AutoModelForTokenClassification.from_pretrained(name_or_path_to_model_folder),args=training_args)