Commit 7ec00c4f authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

ban numpy multithreading

parent c4ae6c7c
......@@ -59,6 +59,8 @@ class Pipeline:
if filter_warnings:
self.filter_warnings()
self.ban_multithreading()
def ban_multithreading(self):
os.environ["OMP_NUM_THREADS"] = "1"
os.environ["MKL_NUM_THREADS"] = "1"
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment