-
Notifications
You must be signed in to change notification settings - Fork 141
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
[ASK] Multithreading in BiVAE #410
Comments
Hi @ip-iulia, the current implementation of BiVAE is itself not Multithreaded, we can therefore not specify the number of threads. However, it is possible that some predefined functions used in BiVAE code are multithreaded, but still currently we do not have control over the number of threads for those functions. |
@ip-iulia the implementation of BiVAE uses PyTorch as backend so it's up to PyTorch to obtain best efficiency based on your available resources. Any particular reason why you want to set a specific number of threads? |
@ip-iulia you can use torch.set_num_threads(int) API. However, it also depends on what parallelization library being used on your machine and needed to be configured properly. More details can be found here. |
Description
I have recently started to experiment with the BiVAE model. I noticed that the training of this model appears to be multithreaded when it is trained on CPU only. Could you please advise if there is a way to control or know the number of threads used by the model? Is there a default number of threads or a parameter that controls this? Thank you.
The text was updated successfully, but these errors were encountered: