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)
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))
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))
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))
def register_ops(self) -> None:
Op.register_op(
'gs_op', lambda op_desc, arch_params, affine: GsOp(
op_desc, arch_params, affine))
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))
def register_ops(self) -> None:
Op.register_op(
'nasbench101_op',
lambda op_desc, arch_params, affine: NasBench101Op(
op_desc, arch_params, affine))
def register_ops(self) -> None:
Op.register_op(
'mixed_op', lambda op_desc, arch_params, affine: MixedOp(
op_desc, arch_params, affine))