Source code for archai.algos.gumbelsoftmax.gs_exp_runner

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

from typing import Type

from overrides import overrides

from archai.nas.exp_runner import ExperimentRunner
from archai.nas.arch_trainer import TArchTrainer
from archai.nas.finalizers import Finalizers
from .gs_model_desc_builder import GsModelDescBuilder
from .gs_arch_trainer import GsArchTrainer
from .gs_finalizers import GsFinalizers

[docs]class GsExperimentRunner(ExperimentRunner):
[docs] @overrides def model_desc_builder(self)->GsModelDescBuilder: return GsModelDescBuilder()
[docs] @overrides def trainer_class(self)->TArchTrainer: return GsArchTrainer
[docs] @overrides def finalizers(self)->Finalizers: return GsFinalizers()