Я хочу получить trainable_variables
из модели, которая создана с помощью подкласса tf.keras.Model
, но она возвращает []
. Я знаю, что это работает при использовании tf.keras.Sequential
.
class MyModel(tf.keras.Model):
def __init__(self):
super(MyModel, self).__init__()
self.dense1 = tf.keras.layers.Dense(4, activation=tf.nn.relu)
self.dense2 = tf.keras.layers.Dense(5, activation=tf.nn.softmax)
self.des = tf.constant([[1.,2.]])
def call(self, inputs):
x = self.dense1(inputs)
# y = self.des
return self.dense2(x)
model = MyModel()
print(model.trainable_variables) #=>[]
Я пробовал подачу некоторые значения для модели, но она все равно возвращает []
.