search

ahmedosman / STAR

ECCV2020 - Official code repository for the paper : STAR - A Sparse Trained Articulated Human Body Regressor
https://star.is.tue.mpg.de
Other
654 stars 95 forks source link

Running without GPU #21

Closed panangam closed 3 years ago

panangam commented 3 years ago

I'm working on a machine without a GPU, and it would be really nice to be able to use STAR without cuda for developing. From my understanding, most of the code imports torch.cuda so when GPU isn't available, it throws an error. Would it be possible to disable that?

ahmedosman commented 3 years ago

Feel free to make a pull request, it will be much appreciated! thanks, Ahmed