Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "7cb203fae4e7964e9e99400b375d660ebce765ee"
Unverified Commit b151ddb9 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix batch size in evaluation loop (#16763) 

* Fix batch size in evaluation loop

* remove debug statement
parent d8269eb4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/trainer.py
View file @ b151ddb9
...@@ -2411,7 +2411,7 @@ class Trainer: ...@@ -2411,7 +2411,7 @@ class Trainer:
elif args.bf16_full_eval: elif args.bf16_full_eval:
model = model.to(dtype=torch.bfloat16, device=args.device) 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} *****") logger.info(f"***** Running {description} *****")
if has_length(dataloader): if has_length(dataloader):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment