Fix batch size in evaluation loop (#16763)

* Fix batch size in evaluation loop * remove debug statement

Fix batch size in evaluation loop (#16763)
* Fix batch size in evaluation loop * remove debug statement
b151ddb9 · Sylvain Gugger · GitHub · d8269eb4 · b151ddb9
Unverified Commit b151ddb9 authored Apr 14, 2022 by Sylvain Gugger Committed by GitHub Apr 14, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/trainer.py src/transformers/trainer.py +1 -1

No files found.
--- a/src/transformers/trainer.py
+++ b/src/transformers/trainer.py
@@ -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):