[ASK] Multithreading in BiVAE

[ip-iulia]

[ASK] Multithreading in BiVAE

[saghiles]

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.

[tqtg]

@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]

Hi @saghiles, @tqtg, thank you for your replies. I was curious to see if the number of threads could be configured rather than allowing PyTorch to determine it based on the available resources.

[tqtg]

@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.