PyTorch Forums
Most efficient way to run model on gpu while optimizing input
autograd
ptrblck
October 11, 2023, 5:26pm
2
You could apply CPU-offloading as described
here
or reduce the number of iterations, if possible.
show post in topic