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]¶
-