如何在Tensorflow 2.0中复制网络

import tensorflow as tf
from tensorflow.keras import layers

class MyModel(tf.keras.Model):
    def __init__(self):
        super(MyModel, self).__init__()
        self.dense1 = layers.Dense(64, activation='relu')
        self.dense2 = layers.Dense(10, activation='softmax')

    def call(self, inputs):
        x = self.dense1(inputs)
        return self.dense2(x)

model = MyModel()

strategy = tf.distribute.MirroredStrategy()

with strategy.scope():
    model = MyModel()
    optimizer = tf.keras.optimizers.Adam()

model.compile(optimizer=optimizer, loss='sparse_categorical_crossentropy', metrics=['accuracy'])

train_dataset = ...
model.fit(train_dataset, epochs=10)