首页 > 解决方案 > Tensorflow 中 Caffe 的 weight_filler 是什么?

问题描述

我正在尝试将 TensorFlow 转换为 Caffe 模型,但在 Caffe 中呢weight_filler?我在 tf 中的模型是:

model.add(Conv2D(16, kernel_size=(5, 5), padding="same", activation='selu', 
                 input_shape=(64, 64, 1)))

model.add(Conv2D(16, kernel_size=(5, 5), padding="same", activation='selu'))

model.add(Conv2D(16, kernel_size=(5, 5), padding="same", activation='selu'))

model.add(Conv2D(16, kernel_size=(5, 5), padding="same", activation='selu'))

model.add(Conv2D(16, kernel_size=(5, 5), padding="same", activation='selu'))

model.add(Conv2D(16, kernel_size=(5, 5), padding="same", activation='selu'))

model.add(Conv2D(16, kernel_size=(5, 5), padding="same", activation='selu'))

model.add(Conv2D(16, kernel_size=(5, 5), padding="same", activation='selu'))

model.add(MaxPooling2D(pool_size=(2, 2),strides=(2,2)))

标签: tensorflowdeep-learningconv-neural-networkcaffe

解决方案


weight_filler是用于初始化权重和偏差的生成器类型。在 tensorflow 中,如果未指定默认初始化程序,则glorot_uniform_initializer它也被调用Xavier uniform initializer,因此 Caffe 中的等效初始化程序为xavier

weight_filler {
  type: "xavier"
}

推荐阅读