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-Bench/main-sss.py
View File

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

4
exps/NATS-Bench/main-tss.py
View File

@@ -340,7 +340,7 @@ def train_single_model(
torch.backends.cudnn.enabled = True
torch.backends.cudnn.deterministic = True
# torch.backends.cudnn.benchmark = True
torch.set_num_threads(workers)
# torch.set_num_threads(workers)
save_dir = (
Path(save_dir)
@@ -675,7 +675,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 if args.workers > 0 else 1)
# torch.set_num_threads(args.workers if args.workers > 0 else 1)
main(
save_dir,