archai.algos.random package

Submodules

archai.algos.random.random_exp_runner module

class archai.algos.random.random_exp_runner.RandomExperimentRunner(config_filename: str, base_name: str, clean_expdir=False)[source]

Bases: archai.nas.exp_runner.ExperimentRunner

model_desc_builder()archai.algos.random.random_model_desc_builder.RandomModelDescBuilder[source]
trainer_class() → Optional[Type[archai.nas.arch_trainer.ArchTrainer]][source]

archai.algos.random.random_model_desc_builder module

class archai.algos.random.random_model_desc_builder.RandOps(n_nodes: int, max_edges: int)[source]

Bases: object

Container to store (op_names, to_states) for each nodes

PRIMITIVES = ['max_pool_3x3', 'avg_pool_3x3', 'skip_connect', 'sep_conv_3x3', 'sep_conv_5x5', 'dil_conv_3x3', 'dil_conv_5x5']
class archai.algos.random.random_model_desc_builder.RandomModelDescBuilder[source]

Bases: archai.nas.model_desc_builder.ModelDescBuilder

build_cells(in_shapes: List[List[List[int]]], conf_model_desc: archai.common.config.Config) → Tuple[List[archai.nas.model_desc.CellDesc], List[Optional[archai.nas.model_desc.AuxTowerDesc]]][source]
build_nodes(stem_shapes: List[List[int]], conf_cell: archai.common.config.Config, cell_index: int, cell_type: archai.nas.model_desc.CellType, node_count: int, in_shape: List[int], out_shape: List[int]) → Tuple[List[List[int]], List[archai.nas.model_desc.NodeDesc]][source]

Module contents