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

Update time_budget for NATS (algos)

Browse Source
This commit is contained in:
D-X-Y
2020-11-26 14:43:28 +08:00
parent 8caf85917b
commit 5456939d81
6 changed files with 106 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -194,7 +194,8 @@ if __name__ == '__main__':
api = create(None, args.search_space, fast_mode=True, verbose=False)
args.save_dir = os.path.join('{:}-{:}'.format(args.save_dir, args.search_space), args.dataset, 'REINFORCE-{:}'.format(args.learning_rate))
args.save_dir = os.path.join('{:}-{:}'.format(args.save_dir, args.search_space),
'{:}-T{:}'.format(args.dataset, args.time_budget), 'REINFORCE-{:}'.format(args.learning_rate))
print('save-dir : {:}'.format(args.save_dir))
if args.rand_seed < 0: