# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from typing import Optional
from overrides import overrides
from archai.nas.exp_runner import ExperimentRunner
from archai.nas.arch_trainer import TArchTrainer
from archai.algos.darts.darts_model_desc_builder import DartsModelDescBuilder
from archai.algos.darts.bilevel_arch_trainer import BilevelArchTrainer
[docs]class DartsExperimentRunner(ExperimentRunner):
[docs] @overrides
def model_desc_builder(self)->DartsModelDescBuilder:
return DartsModelDescBuilder()
[docs] @overrides
def trainer_class(self)->TArchTrainer:
return BilevelArchTrainer