MatthiasPicard commited on
Commit
be34aad
·
verified ·
1 Parent(s): 72eea34

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +1 -1
tasks/text.py CHANGED
@@ -84,7 +84,7 @@ async def evaluate_text(request: TextEvaluationRequest):
84
  tokenizer=tokenizer
85
  )
86
 
87
- trainer.args.per_device_eval_batch_size = 2
88
 
89
  preds = trainer.predict(tokenized_test)
90
 
 
84
  tokenizer=tokenizer
85
  )
86
 
87
+ trainer.args.per_device_eval_batch_size = 4
88
 
89
  preds = trainer.predict(tokenized_test)
90