Tensorflow: что эквивалентно weight_filler из Caffe в Tensorflow?

Я пытаюсь преобразовать TensorFlow в Caffemodel, но как насчет weight_filler в Caffe? Моя модель в tf:

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

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

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

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

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

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

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

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

model.add(MaxPooling2D (размер_пул = (2, 2), шаги = (2,2)))

1 ответ

Решение

Weight _filler - это тип генератора, используемый для инициализации весов и смещений. В тензорном потоке, если он не указан, инициализатором по умолчанию являетсяglorot_uniform_initializer который также называют Xavier uniform initializer поэтому эквивалентный инициализатор в Caffe xavier:

weight_filler {
  type: "xavier"
}
Другие вопросы по тегам