Skip to content
Snippets Groups Projects
Commit f59f58b7 authored by Yannick DAYER's avatar Yannick DAYER
Browse files

Set torch to use one thread. Fixes a bug on docker

parent d35f4f73
No related branches found
No related tags found
1 merge request!135Set torch to use one thread
Pipeline #65526 passed
......@@ -13,6 +13,8 @@ from bob.io.image import to_matplotlib
logger = logging.getLogger(__name__)
# Fix for torch on docker (fixes CI linux runners):
torch.set_num_threads(1)
DEEP_PIX_BIS_PRETRAINED_MODELS = {
"oulu-npu-p1": [
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment