Place inputs on device when include_inputs_for_metrics is True #18046
Place inputs on device when include_inputs_for_metrics is True
448a509b
sgugger
merged
1b5ea747
into main 3 years ago
sgugger
deleted the input_trainer branch 3 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub