TensorFlow1.14相当于TensorFlow2.0 model.trainable_variables

Question

如何替换Tensorflow.1.14 (https://www.tensorflow.org/beta/guide/effective_tf2)中的TensorFlow2.0 model.trainable_variables？

Answer 1

假设您有两个tf.keras.Model

model_1 = tf.keras.Sequential([tf.keras.layers.Dense(100, name="model_1/dense1")])
model_1.build(input_shape=(None, 50))
model_2 = tf.keras.Sequential([tf.keras.layers.Dense(100, name="model_2/dense1")])
model_2.build(input_shape=(None, 50))

您可以像这样获得第一个模型的可训练变量：

model_1_trainable = [variable for variable in tf.trainable_variables() if variable.name.startswith("model_1")]