Как я могу получить trainable_variables из подкласса модели tf.keras.Model? - PullRequest
0 голосов
/ 15 марта 2020

Я хочу получить 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)  #=>[]

Я пробовал подачу некоторые значения для модели, но она все равно возвращает [].

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...