Есть ли способ заменить функцию PReLU на ReLU в кафе с помощью операции net? - PullRequest
Новая
       87

Есть ли способ заменить функцию PReLU на ReLU в кафе с помощью операции net?

0 голосов
/ 17 марта 2020

У меня есть следующая конфигурация сети: 

layer {
  name: "conv4_5"
  type: "Convolution"
  bottom: "conv4_4"
  top: "conv4_5"
  param {
    lr_mult: 1
    decay_mult: 1
  }
  param {
    lr_mult: 0
    decay_mult: 0
  }
  convolution_param {
    num_output: 512
    kernel_size: 3
    stride: 1
    pad: 1
    weight_filler {
      type: "gaussian"
      std: 0.01
    }
    bias_filler {
      type: "constant"
      value: 0
    }
  }
}
layer {
  name: "bn4_5"
  type: "BatchNorm"
  bottom: "conv4_5"
  top: "conv4_5"
}
layer {
    bottom: "conv4_5"
    top: "conv4_5"
    name: "scale_conv4_5"
    type: "Scale"
    scale_param {
        bias_term: true
    }
}
layer {
  name: "relu4_5"
  type: "PReLU"
  bottom: "conv4_5"
  top: "conv4_5"
}

По сути, я хочу заменить 

layer {
  name: "relu4_5"
  type: "PReLU"
  bottom: "conv4_5"
  top: "conv4_5"
}

на leaky-relu 

layer {
  name: "relu4_5"
  type: "ReLU"
  bottom: "conv4_5"
  top: "conv4_5"
relu_param{
negative_slope: 0.01
}
}

с net -хирургическим инструментом кафе. Может ли кто-нибудь помочь мне с этим?

Усилия с моей стороны: я уже изучил этот ответ: ответ с переполнением стека , но не смог получить окончательный результат. Большое спасибо.

...