Example #1
0
from archai.common.common import common_init
from archai.nas.evaluate import eval_arch
from archai.petridish.petridish_micro_builder import PetridishMicroBuilder

if __name__ == '__main__':
    conf = common_init(
        config_filepath='confs/petridish_cifar.yaml',
        param_args=['--common.experiment_name', 'petridish_cifar_eval'])

    conf_eval = conf['nas']['eval']

    # evaluate architecture using eval settings
    eval_arch(conf_eval, micro_builder=PetridishMicroBuilder())

    exit(0)
Example #2
0
from archai.darts.darts_micro_builder import DartsMicroBuilder
from archai.common.common import common_init
from archai.nas.evaluate import eval_arch

if __name__ == '__main__':
    conf = common_init(
        config_filepath='confs/darts_cifar.yaml',
        param_args=['--common.experiment_name', 'darts_cifar_eval'])

    conf_eval = conf['nas']['eval']

    # evaluate architecture using eval settings
    eval_arch(conf_eval, micro_builder=DartsMicroBuilder())

    exit(0)
Example #3
0
import os

from archai.common.common import common_init
from archai.nas.evaluate import eval_arch
from archai.random_arch.random_micro_builder import RandomMicroBuilder

from archai.nas import nas_utils

if __name__ == '__main__':
    conf = common_init(
        config_filepath='confs/random_cifar.yaml',
        param_args=['--common.experiment_name', 'random_cifar_eval'])

    # region config
    conf_eval = conf['nas']['eval']
    final_desc_filename = conf_eval['final_desc_filename']
    # endregion

    # evaluate architecture using eval settings
    eval_arch(conf_eval, micro_builder=RandomMicroBuilder())

    exit(0)

# LOG_DIR = os.environ.get('PT_OUTPUT_DIR', '')
# DATA_DIR = os.environ.get('PT_DATA_DIR', '')
Example #4
0
from archai.common.common import common_init
from archai.darts.darts_micro_builder import DartsMicroBuilder
from archai.nas.evaluate import eval_arch

if __name__ == '__main__':
    conf = common_init(
        config_filepath='confs/imagenet_darts.yaml',
        param_args=['--common.experiment_name', 'darts_imagenet_eval'])

    conf_eval = conf['nas']['eval']

    micro_builder = DartsMicroBuilder()
    eval_arch(conf_eval, micro_builder=micro_builder)

    exit(0)
Example #5
0
 def _run_eval(self, conf_eval: Config) -> None:
     evaluate.eval_arch(conf_eval, cell_builder=self.cell_builder())