From 513581208e099f9bccb9497cc45365e129a84a1e Mon Sep 17 00:00:00 2001
From: Engin Eren <engin.eren@desy.de>
Date: Wed, 6 Apr 2022 17:09:00 +0200
Subject: [PATCH] MKL error

---
 wgan.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/wgan.py b/wgan.py
index 95cf6db..19360a9 100644
--- a/wgan.py
+++ b/wgan.py
@@ -12,6 +12,8 @@ from torch.utils.data import DataLoader
 from torch.autograd import Variable
 
 
+os.environ['MKL_THREADING_LAYER'] = 'GNU'
+
 torch.autograd.set_detect_anomaly(True)
 
 sys.path.append('/opt/regressor/src')
-- 
GitLab