-
Notifications
You must be signed in to change notification settings - Fork 26
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 optimize CPU usage by modifying test_EDTalk_A_use_exp_weight function to leverage GPU #31
Comments
I have checked the codes, and the current models and function calculation are on GPU. I think only loading data depends on CPU. |
I noticed a significant increase in CPU usage in the following part of demo_EDTalk_A_using_predefined_exp_weights.py
I suspect that the function test_EDTalk_A_use_exp_weight in generator.py is causing this spike in CPU usage. I am planning to modify this function to utilize GPU (CUDA) more efficiently in order to reduce the load on the CPU and improve the calculation speed, anyone here?
The text was updated successfully, but these errors were encountered: