Source code for archai.algos.manual.manual_exp_runner

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


from typing import Optional, Type

from overrides import overrides

from archai.common.config import Config
from archai.nas import nas_utils
from archai.nas.exp_runner import ExperimentRunner
from archai.nas.arch_trainer import ArchTrainer, TArchTrainer
from archai.nas.model_desc_builder import ModelDescBuilder
from archai.algos.manual.manual_searcher import ManualSearcher
from archai.algos.manual.manual_evaluater import ManualEvaluater


[docs]class ManualExperimentRunner(ExperimentRunner): """Runs manually designed models such as resnet"""
[docs] @overrides def model_desc_builder(self)->Optional[ModelDescBuilder]: return None
[docs] @overrides def trainer_class(self)->TArchTrainer: return None # no search trainer
[docs] @overrides def searcher(self)->ManualSearcher: return ManualSearcher()
[docs] @overrides def evaluater(self)->ManualEvaluater: return ManualEvaluater()
[docs] @overrides def copy_search_to_eval(self)->None: pass