Merge pull request #21 from ethancaballero/patch-4

automate OMP_NUM_THREADS=1 setting
stan-stripe · May 27, 2017 · 842ec4d · 842ec4d
2 parents db57a5e + d70e677
commit 842ec4d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -40,6 +40,8 @@
 
 
 if __name__ == '__main__':
+    os.environ['OMP_NUM_THREADS'] = '1'  
+
     args = parser.parse_args()
 
     torch.manual_seed(args.seed)