Skip to content

Commit 21dde0d

Browse files
committed
1SHOT1 MO version
1 parent c2f1454 commit 21dde0d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

hpobench/benchmarks/nas/nasbench_1shot1.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -598,7 +598,7 @@ class NASBench1shot1SearchSpace1Benchmark(NASBench1shot1BaseBenchmark):
598598
def __init__(self, data_path: Union[Path, str, None] = None,
599599
rng: Union[np.random.RandomState, int, None] = None):
600600
super(NASBench1shot1SearchSpace1Benchmark, self).__init__(data_path=data_path, rng=rng)
601-
self.mo_benchmark = NASBench1shot1SearchSpace1Benchmark(data_path=data_path, rng=rng)
601+
self.mo_benchmark = NASBench1shot1SearchSpace1MOBenchmark(data_path=data_path, rng=rng)
602602

603603
@staticmethod
604604
def get_configuration_space(seed: Union[int, None] = None) -> CS.ConfigurationSpace:
@@ -609,7 +609,7 @@ class NASBench1shot1SearchSpace2Benchmark(NASBench1shot1BaseBenchmark):
609609
def __init__(self, data_path: Union[Path, str, None] = None,
610610
rng: Union[np.random.RandomState, int, None] = None):
611611
super(NASBench1shot1SearchSpace2Benchmark, self).__init__(data_path=data_path, rng=rng)
612-
self.mo_benchmark = NASBench1shot1SearchSpace2Benchmark(data_path=data_path, rng=rng)
612+
self.mo_benchmark = NASBench1shot1SearchSpace2MOBenchmark(data_path=data_path, rng=rng)
613613

614614
@staticmethod
615615
def get_configuration_space(seed: Union[int, None] = None) -> CS.ConfigurationSpace:
@@ -620,7 +620,7 @@ class NASBench1shot1SearchSpace3Benchmark(NASBench1shot1BaseBenchmark):
620620
def __init__(self, data_path: Union[Path, str, None] = None,
621621
rng: Union[np.random.RandomState, int, None] = None):
622622
super(NASBench1shot1SearchSpace3Benchmark, self).__init__(data_path=data_path, rng=rng)
623-
self.mo_benchmark = NASBench1shot1SearchSpace3Benchmark(data_path=data_path, rng=rng)
623+
self.mo_benchmark = NASBench1shot1SearchSpace3MOBenchmark(data_path=data_path, rng=rng)
624624

625625
@staticmethod
626626
def get_configuration_space(seed: Union[int, None] = None) -> CS.ConfigurationSpace:

0 commit comments

Comments
 (0)