Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	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...
         | 

