Skip to content
Snippets Groups Projects

Unset the number of used threads

1 file
+ 0
3
Compare changes
  • Side-by-side
  • Inline
@@ -23,9 +23,6 @@ from bob.bio.face.utils import (
embedding_transformer,
)
# Fix for torch on docker (fixes CI linux runners):
torch.set_num_threads(1)
class PyTorchModel(TransformerMixin, BaseEstimator):
"""
Loading