예제 #1
0
    worker = worker(min_budget=min_budget,
                    max_budget=max_budget,
                    eta=eta,
                    search_space=args.space,
                    nasbench_data=nasbench,
                    seed=args.seed,
                    nameserver=ns_host,
                    nameserver_port=ns_port,
                    run_id=args.run_id)
    worker.run(background=True)

    #instantiate BOHB and run it
    result_logger = hputil.json_result_logger(directory=args.working_directory,
                                              overwrite=True)

    HPB = BOHB(configspace=worker.get_config_space(),
               working_directory=args.working_directory,
               run_id=args.run_id,
               eta=eta,
               min_budget=min_budget,
               max_budget=max_budget,
               host=ns_host,
               nameserver=ns_host,
               nameserver_port=ns_port,
               ping_interval=3600,
               result_logger=result_logger)

    res = HPB.run(n_iterations=args.num_iterations,
                  min_n_workers=args.total_num_workers)

    with open(os.path.join(args.working_directory, 'results.pkl'), 'wb') as fh:
예제 #2
0
                    search_space=args.space,
                    algorithm=args.algorithm,
                    nasbench_data=nasbench,
                    seed=args.seed,
                    unrolled=args.unrolled,
                    nameserver=ns_host,
                    nameserver_port=ns_port,
                    run_id=args.run_id)
    worker.run(background=True)

    #instantiate BOHB and run it
    result_logger = hputil.json_result_logger(directory=args.working_directory,
                                              overwrite=True)

    HPB = BOHB(
        configspace=worker.get_config_space(),
        working_directory=args.working_directory,
        run_id=args.run_id,
        eta=eta,
        min_budget=min_budget,
        max_budget=max_budget,
        host=ns_host,
        nameserver=ns_host,
        nameserver_port=ns_port,
        ping_interval=3600,
        result_logger=result_logger,
        start_from_default=True,
    )

    res = HPB.run(n_iterations=args.num_iterations,
                  min_n_workers=args.total_num_workers)