How to modify batch normalization layers (DeconvNet) to be able to run with caffe?

后端 未结 1 830
别跟我提以往
别跟我提以往 2021-01-23 08:17

I wanted to run the Deconvnet on my data, however it seemd it has been written for another version of caffe. Does anyone know how to change batch_params

1条回答
  •  借酒劲吻你
    2021-01-23 09:04

    Could you please let me know if this is correct to change like this?

    layer {
      name: "bn1_1"
      type: "BatchNorm"
      bottom: "conv1_1"
      top: "conv1_1"
      param {
        lr_mult: 0
      }
      param {
        lr_mult: 0
      }
      param {
        lr_mult: 0
      }
      include {
        phase: TEST
      }
      batch_norm_param {
        use_global_stats: true
      }
    }
    layer {
      name: "scale_conv1_1"
      type: "Scale"
      bottom: "conv1_1"
      top: "conv1_1"
      scale_param {
        bias_term: true
        bias_filler {
          type: "constant"
          value: 0.001
        }
      }
    }
    

    0 讨论(0)
提交回复
热议问题