如何在Tensorflow 2.0中复制网络

Question

I am not sure on how to copy over a network in Tensorflow 2.0. There are plenty of answers on how to do it in Tensorflow 1.x, but none about 2.0. Both of the networks are made through subclassing the tf.keras.Model , so I can't use the tf.keras.models.clone_model function.

I have tried different approaches outlined below but none of them seem to work.

network1 = network2
network1.weights = network2.weights

from copy import copy
network1 = copy(network2)

Some of these methods will make a reference to the current network but not actually copy it. Would appreciate all the help I can get!

Answer 1

Suppose that model_a and model_b are instantiations of the same Keras Model. Then do:

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