Source code for archai.supergraph.algos.nasbench101.nasbench101_exp_runner

# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.

from overrides import overrides

from archai.supergraph.algos.petridish.petridish_model_desc_builder import (
    PetridishModelBuilder,
)
from archai.supergraph.nas.arch_trainer import ArchTrainer, TArchTrainer
from archai.supergraph.nas.exp_runner import ExperimentRunner


[docs]class PetridishExperimentRunner(ExperimentRunner):
[docs]    @overrides
    def model_desc_builder(self)->PetridishModelBuilder:
        return PetridishModelBuilder()

[docs]    @overrides
    def trainer_class(self)->TArchTrainer:
        return ArchTrainer