Разделение веса в Керасе
Я пытаюсь использовать один и тот же вес для двух слоев, которые в нижеследующем коде названы плотность3. Я знаю, что использование ' density3 ' в качестве имени не допускается, поскольку слои должны иметь уникальные имена. Что лучше всего было указать одинаковый вес разных слоев в керасе
model = keras.Sequential([
keras.layers.Flatten(input_shape=(28, 28)),
keras.layers.Dense(128, activation=tf.nn.relu,name='dense_1'),
keras.layers.Dense(10, activation=tf.nn.relu, name='dense_2'),
keras.layers.Dense(10, activation=tf.nn.relu,name='dense_3' ),
keras.layers.Dense(10, activation=tf.nn.relu,name='dense_3' ),
keras.layers.Dense(10, activation=tf.nn.softmax)
])
1 ответ
sharedLayer = keras.layers.Dense(10, activation=tf.nn.relu,name='dense_3' )
model = keras.Sequential([
keras.layers.Flatten(input_shape=(28, 28)),
keras.layers.Dense(128, activation=tf.nn.relu,name='dense_1'),
keras.layers.Dense(10, activation=tf.nn.relu, name='dense_2'),
sharedLayer,
sharedLayer,
keras.layers.Dense(10, activation=tf.nn.softmax)
])