Refine lib -> xautodl

2021-05-19 08:10:42 +00:00
parent bd407ac4dc
commit 1c6c3e7166
12 changed files with 83 additions and 53 deletions
--- a/exps/NATS-algos/reinforce.py
+++ b/exps/NATS-algos/reinforce.py
@@ -132,7 +132,7 @@ def select_action(policy):


 def main(xargs, api):
-    torch.set_num_threads(4)
+    # torch.set_num_threads(4)
    prepare_seed(xargs.rand_seed)
    logger = prepare_logger(args)