与Tensorflow 2.0模型等效的Tensorflow 1.14.trainable_variables

Question

如何在Tensorflow.1.14（https://www.tensorflow.org/beta/guide/effective_tf2）中替换Tensorflow 2.0 __dict__？

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")]