def test_kernel_reg(case): x, y, model_fn, backbone= case l1_reg = keras.regularizers.l1(0.1) model = model_fn(backbone) reg_model = set_regularization(model, kernel_regularizer=l1_reg) _test_regularizer(model, reg_model, x, y) l2_reg = keras.regularizers.l2(0.1) model = model_fn(backbone, encoder_weights=None) reg_model = set_regularization(model, kernel_regularizer=l2_reg) _test_regularizer(model, reg_model, x, y)
def test_activity_reg(case): x, y, model_fn, backbone= case l2_reg = keras.regularizers.l2(1) model = model_fn(backbone) reg_model = set_regularization(model, activity_regularizer=l2_reg) _test_regularizer(model, reg_model, x, y)
def test_bn_reg(case): x, y, model_fn, backbone= case l1_reg = keras.regularizers.l1(1) model = model_fn(backbone) reg_model = set_regularization(model, gamma_regularizer=l1_reg) _test_regularizer(model, reg_model, x, y) model = model_fn(backbone) reg_model = set_regularization(model, beta_regularizer=l1_reg) _test_regularizer(model, reg_model, x, y) l2_reg = keras.regularizers.l2(1) model = model_fn(backbone) reg_model = set_regularization(model, gamma_regularizer=l2_reg) _test_regularizer(model, reg_model, x, y) model = model_fn(backbone) reg_model = set_regularization(model, beta_regularizer=l2_reg) _test_regularizer(model, reg_model, x, y)