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)
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)
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', '')
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)
def _run_eval(self, conf_eval: Config) -> None: evaluate.eval_arch(conf_eval, cell_builder=self.cell_builder())