Update time_budget for NATS (algos)

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

View File

@@ -73,7 +73,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, 'RANDOM')
args.save_dir = os.path.join('{:}-{:}'.format(args.save_dir, args.search_space),
'{:}-T{:}'.format(args.dataset, args.time_budget), 'RANDOM')
print('save-dir : {:}'.format(args.save_dir))
if args.rand_seed < 0: