Tensorflow 2.0 如何在卷积层之间共享参数？

Question

我正在尝试在 Tensorflow 2.0 中重新实现 Multi-View CNN (MVCNN)。但是，据我所见，keras 层没有 tf.layers 中的选项reuse=True|False。有什么方法可以定义使用新 API 共享参数的层？或者我需要以 TFv1 方式构建模型？

非常感谢！

Answer 1

要共享模型的参数，您只需使用相同的模型。这是 TensorFlow 2.0 中引入的新范式； 在 TF 1.xt 中，我们使用了面向图的方法，我们需要重复使用同一个图来共享变量，但现在我们可以重复使用具有不同输入的同一个 tf.keras.Model 对象。

是携带自己变量的对象。

使用 Keras 模型和tf.GradientTape，您可以轻松地训练一个共享变量的模型，如下例所示。

# This is your model definition
model = tf.keras.Sequential(...)

#input_1,2 are model different inputs

with tf.GradientTape() as tape:
  a = model(input_1)
  b = model(input_2)
  # you can just copute the loss
  loss = a + b

# Use the tape to compute the gradients of the loss
# w.r.t. the model trainable variables

grads = tape.gradient(loss, model.trainable_varibles)

# Opt in an optimizer object, like tf.optimizers.Adam
# and you can use it to apply the update rule, following the gradient direction

opt.apply_gradients(zip(grads, model.trainable_variables))