Python Op.register_op Examples

Example #1

File: nasbench101_model_desc_builder.py Project: singlasahil14/archai

    def pre_build(self, conf_model_desc:Config)->None:
        Op.register_op('nasbench101_op',
                       lambda op_desc, arch_params, affine:
                           NasBench101Op(op_desc, arch_params, affine))

        # extract model specs from params in config
        params = conf_model_desc['params'].to_dict()
        cell_matrix = params['cell_matrix']
        vertex_ops = params['vertex_ops']
        self.num_stacks = params['num_stacks']

        self._cell_matrix, self._vertex_ops = model_matrix.prune(cell_matrix, vertex_ops)

Example #2

 def pre_build(self, conf_model_desc: Config) -> None:
     Op.register_op(
         'petridish_normal_op',
         lambda op_desc, arch_params, affine: PetridishOp(
             op_desc, arch_params, False, affine))
     Op.register_op(
         'petridish_reduction_op',
         lambda op_desc, arch_params, affine: PetridishOp(
             op_desc, arch_params, True, affine))
     Op.register_op(
         'temp_identity_op',
         lambda op_desc, arch_params, affine: TempIdentityOp(op_desc))

Example #3

File: petridish_cell_builder.py Project: the-darklord/archai

 def register_ops(self) -> None:
     Op.register_op(
         'petridish_normal_op',
         lambda op_desc, arch_params, affine: PetridishOp(
             op_desc, arch_params, False, affine))
     Op.register_op(
         'petridish_reduction_op',
         lambda op_desc, arch_params, affine: PetridishOp(
             op_desc, arch_params, True, affine))
     Op.register_op(
         'temp_identity_op',
         lambda op_desc, arch_params, affine: TempIdentityOp(op_desc))

Example #4

File: divnas_model_desc_builder.py Project: wayne9qiu/archai

 def pre_build(self, conf_model_desc: Config) -> None:
     Op.register_op(
         'div_op', lambda op_desc, arch_params, affine: DivOp(
             op_desc, arch_params, affine))

Example #5

 def register_ops(self) -> None:
     Op.register_op(
         'gs_op', lambda op_desc, arch_params, affine: GsOp(
             op_desc, arch_params, affine))

Example #6

 def pre_build(self, conf_model_desc: Config) -> None:
     Op.register_op(
         'nasbench101_op',
         lambda op_desc, arch_params, affine: NasBench101Op(
             op_desc, arch_params, affine))

Example #7

 def register_ops(self) -> None:
     Op.register_op(
         'nasbench101_op',
         lambda op_desc, arch_params, affine: NasBench101Op(
             op_desc, arch_params, affine))

Example #8

 def register_ops(self) -> None:
     Op.register_op(
         'mixed_op', lambda op_desc, arch_params, affine: MixedOp(
             op_desc, arch_params, affine))