shap-e icon indicating copy to clipboard operation
shap-e copied to clipboard
verified publisher icon openai

optimized the code for GPU

Open sachinspanicker opened this issue 2 years ago • 1 comments

sachinspanicker avatar May 10 '23 17:05 sachinspanicker

Optimized the code for GPU

sachinspanicker avatar May 10 '23 17:05 sachinspanicker

The model is already created on the GPU, and so is the data batch. Note that the device variable is already created in an earlier cell and passed to various functions. I believe every to(device) in your diff is redundant.

unixpickle avatar Jul 11 '23 16:07 unixpickle