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-Bench/main-sss.py
+++ b/exps/NATS-Bench/main-sss.py
@@ -470,7 +470,7 @@ if __name__ == "__main__":
    assert torch.cuda.is_available(), "CUDA is not available."
    torch.backends.cudnn.enabled = True
    torch.backends.cudnn.deterministic = True
-    torch.set_num_threads(args.workers)
+    # torch.set_num_threads(args.workers)

    main(
        save_dir,