python - 如何在 Tensorflow 2.0 中复制网络

Question

我不确定如何在 Tensorflow 2.0 中通过网络进行复制。关于如何在 Tensorflow 1.x 中做到这一点有很多答案，但没有关于 2.0 的答案。这两个网络都是通过子类化 tf.keras.Model 创建的，因此我无法使用 tf.keras.models.clone_model 函数。

我尝试了下面列出的不同方法，但似乎都不起作用。

network1 = network2
network1.weights = network2.weights

from copy import copy
network1 = copy(network2)

其中一些方法将引用当前网络，但实际上并不复制它。感谢我能得到的所有帮助!

Answer 1

假设 model_a 和 model_b 是同一 Keras 模型的实例。然后执行:

for a, b in zip(model_a.variables, model_b.variables):
  a.assign(b)  # copies the variables of model_b into model_a