iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refine lib -> xautodl

Browse Source
This commit is contained in:
D-X-Y
2021-05-19 08:10:42 +00:00
parent bd407ac4dc
commit 1c6c3e7166
12 changed files with 83 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
exps/NATS-algos/reinforce.py
View File

@@ -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)

2
exps/NATS-algos/search-size.py
View File

@@ -204,7 +204,7 @@ def main(xargs):
torch.backends.cudnn.enabled = True
torch.backends.cudnn.benchmark = False
torch.backends.cudnn.deterministic = True
torch.set_num_threads(xargs.workers)
# torch.set_num_threads(xargs.workers)
prepare_seed(xargs.rand_seed)
logger = prepare_logger(args)