maybe fix a problem with threadpoolctl

Giammarco07 · Sep 16, 2019 · 52c4c77 · 52c4c77
1 parent 3ff4541
commit 52c4c77
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/batchgenerators/dataloading/multi_threaded_augmenter.py b/batchgenerators/dataloading/multi_threaded_augmenter.py
@@ -33,7 +33,7 @@
 
 def producer(queue, data_loader, transform, thread_id, seed, abort_event):
     try:
-        with threadpool_limits(limits=1):
+        with threadpool_limits(limits=1, user_api="blas"):
             np.random.seed(seed)
             data_loader.set_thread_id(thread_id)
             item = None