-
Notifications
You must be signed in to change notification settings - Fork 227
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
How to run CenterHMR on CPU? #20
Comments
Thanks for your interests! For details, please check the last two commits: |
Thanks for the quick reply @Arthur151 if '-1' not in self.gpu (file: src/core/base.py , line no: 41,55,189) I changed this line to " if self.gpu!=-1 " then the above error is gone but got this error: "RuntimeError: No CUDA GPUs are available" Sharing with you a screenshot : https://drive.google.com/file/d/1v9kjrOdI8nXWgjSIK21Zpp-qjUzrRYcm/view?usp=sharing |
There are several things need to be noticed.
|
Yes. I already have taken care of both of these things. I setup the repository in totally new colab file and also put space between GPUS: and -1 (You can see that in the above screenshot , on right side I've kept open the yml file for your reference ). Please take note of this : type of self.gpu is string not int that's why the if condition will be if self.gpu!='-1' not if self.gpu!=-1 Also, the model is running well upto here "finished build model" and after that the error is coming in file test.py and line no:20 : self.visualizer = Visualizer(model_type=self.model_type,resolution=self.vis_size, input_size=self.input_size,with_renderer=True) Error: ValueError: Invalid device ID (0) |
Sorry for the bug. I have just try the Colab. It turns out that the pyrender need GPU for visualization. So current code could run on CPU and save the 3D body mesh / SMPL parameter files. But it won't render the estimated 3D meshes back to 2D image due to the limitation of Pyrender package. In the future, I will try to replace the visualization module to achieve full CPU support. The bug has been fixed. It passed the test on Colab (the last row). |
Don't hesitate to let me know your problem. |
Sure ! |
We have relase a new version with more funny things. Hope you enjoy it. |
Training code has been released along with all evaluation. |
@JANVI2411 |
First of all thanks @Arthur151. I'm able to generate 3D meshes for my videos however I'm curious to know if we can also run this on CPU. it would be very kind of you if you can guide me on this.
The text was updated successfully, but these errors were encountered: