transformers
Place inputs on device when include_inputs_for_metrics is True
#18046
Merged

Place inputs on device when include_inputs_for_metrics is True #18046

sgugger merged 1 commit into main from input_trainer
sgugger
sgugger Place inputs on device when include_inputs_for_metrics is True
448a509b
sgugger sgugger requested a review from LysandreJik LysandreJik 3 years ago
HuggingFaceDocBuilderDev
LysandreJik
LysandreJik approved these changes on 2022-07-07
sgugger sgugger merged 1b5ea747 into main 3 years ago
sgugger sgugger deleted the input_trainer branch 3 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone