transformers
Trainer - add cache clearing and the option for batched eval metrics computation
#28769
Merged
Merged
Trainer - add cache clearing and the option for batched eval metrics computation #28769
muellerzr
merged
20 commits into
huggingface:main
from
FoamoftheSea:trainer-updates
Added cache clearing for GPU efficiency.
99e94ab1
Added cache clearing for GPU efficiency.
fc570304
Added batch_eval_metrics capability
548a26f4
Merge branch 'main' into trainer-updates
fa690a8d
Ran make fixup
f6350aaf
Merge remote-tracking branch 'origin/trainer-updates' into trainer-up…
006ecd23
Fixed bug
786046e0
Fixed whitespace issue
7752d130
Merge branch 'main' into trainer-updates
e155b0b4
Merge branch 'main' into trainer-updates
a26de727
FoamoftheSea
commented
on 2024-02-22
Fixed outdated condition
a12e46f8
Merge branch 'main' into trainer-updates
137bd392
SunMarc
commented
on 2024-02-22
muellerzr
commented
on 2024-02-29
Merge branch 'main' into trainer-updates
994e7d06
muellerzr
approved these changes
on 2024-04-25
muellerzr
commented
on 2024-04-25
SunMarc
approved these changes
on 2024-04-25
Updated docstrings with instructions for batch_eval_metrics. Updated …
9b7c26e8
Added first version of batch_eval_metrics Trainer test
760af34d
Merge branch 'main' into trainer-updates
4f87e817
Fixed batch_eval_metrics Trainer tests for both eval and predict
1fed1bcb
Fixed batch_eval_metrics behavior for new Trainer variables
5f44cb17
Fixed batch_eval_metrics Trainer tests
583d14b1
Ran fixup
f925e8c4
LysandreJik
approved these changes
on 2024-05-06
muellerzr
approved these changes
on 2024-05-06
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub