optimum-benchmark icon indicating copy to clipboard operation
optimum-benchmark copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

set "use_cpu" to true in cpu pytorch training benchmark tests (#159)

Open aliabdelkader opened this issue 11 months ago • 0 comments

  • set training_arguments.use_cpu to true so that the Trainer class runs training on the cpu instead of utilizing the available gpu

aliabdelkader avatar Mar 14 '24 15:03 aliabdelkader