Update plots for NATS-Bench

2020-11-27 02:15:09 +08:00
parent 3fb3222e82
commit 550d24ec07
3 changed files with 43 additions and 60 deletions
--- a/exps/NATS-algos/bohb.py
+++ b/exps/NATS-algos/bohb.py
@@ -167,7 +167,7 @@ if __name__ == '__main__':
  parser.add_argument('--rand_seed',          type=int,  default=-1, help='manual seed')
  args = parser.parse_args()
  
-  api = create(None, args.search_space, fast_mode=True, verbose=False)
+  api = create(None, args.search_space, fast_mode=False, verbose=False)

  args.save_dir = os.path.join('{:}-{:}'.format(args.save_dir, args.search_space),
                               '{:}-T{:}'.format(args.dataset, args.time_budget), 'BOHB')