Fixing multiprocessing issue on some systems

devendrachaplot · Apr 16, 2018 · 3b5891f · 3b5891f
1 parent 681c5ec
commit 3b5891f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/a3c_main.py b/a3c_main.py
@@ -1,6 +1,7 @@
 import vizdoom
 import argparse
 import os
+os.environ["OMP_NUM_THREADS"] = "1"
 import numpy as np
 import torch
 import torch.multiprocessing as mp
@@ -12,8 +13,6 @@
 
 import logging
 
-os.environ["OMP_NUM_THREADS"] = "1"
-
 parser = argparse.ArgumentParser(description='Gated-Attention for Grounding')
 
 # Environment arguments