#273·optimate

[Speedster] `optimize_model` function transfers the original model from the GPU to the CPU

Author: mfumanelliCreated Mar 17, 2023Updated Oct 8, 2023
Labelsgood first issuespeedster

Problem

The optimize_model function transfers the original model from the GPU to the CPU:

image

Desired Behavior

Maintaining the model on the initial device.