Как инициализировать веса в tenorflow.contrib.layers conv2d

Question

Я написал простой автоэнкодер, использующий библиотеку tenorflow. Это пример кода, который я написал для сети автоэнкодера Чего я не понимаю, так это как ввести вес в эту модель, так как она работает сейчас без какой-либо конкретной инициализации?

Answer 1

Инициализация веса определяется при создании слоя conv2d. В API (https://www.tensorflow.org/api_docs/python/tf/contrib/layers/conv2d) в такой функции есть параметр с именем weights_initializer, который может быть заполнен объектом tf.initializer и определяет способ инициализации весов. По умолчанию для TFv1.8 это xavier_initializer (https://www.tensorflow.org/api_docs/python/tf/contrib/layers/xavier_initializer),, но вы можете установить его для любого из включенных здесь инициализаторов: https://www.tensorflow.org/api_docs/python/tf/initializers. Например, если вы хотите, чтобы инициализация веса была постоянной со значением 12345, вы должны написать:

lays.conv2d(inputs, 32,[5,5], stride=2, padding="SAME", 
                   weights_initializer=tf.constant_initializer(12345))