riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 03:01:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix batch size in evaluation loop (#16763)

Browse Source 
* Fix batch size in evaluation loop

* remove debug statement
This commit is contained in:
Sylvain Gugger 2022-04-14 09:22:54 -04:00 committed by GitHub
parent d8269eb4d5
commit b151ddb9b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/trainer.py
View File

@ -2411,7 +2411,7 @@ class Trainer:
elif args.bf16_full_eval:
model = model.to(dtype=torch.bfloat16, device=args.device)
batch_size = self.args.per_device_eval_batch_size
batch_size = self.args.eval_batch_size
logger.info(f"***** Running {description} *****")
if has_length(dataloader):