We should add a feature to our training notebooks to check if an environment variable is set (e.g. KINOML_NUM_THREADS), and if so, call `torch.set_num_threads() with the appropriate number of threads. This will allow us to better work within batch queue systems.
We should add a feature to our training notebooks to check if an environment variable is set (e.g.
KINOML_NUM_THREADS
), and if so, call `torch.set_num_threads() with the appropriate number of threads. This will allow us to better work within batch queue systems.