Настройка функции fit () в keras приводит к тому, что функция Assessment () возвращает пустой список

Question

Я тренирую VAE с помощью учебника Keras: https://keras.io/examples/generative/vae/. Это включает в себя создание класса VAE и определение пользовательской процедуры обучения, которая подробно описана здесь: https://keras.io/guides/customizing_what_happens_in_fit/. После создания кодировщика и декодера, как указано в руководстве, и обучения модели, я создаю модель VAE следующим образом:

vae = VAE(encoder, decoder) 
vae.compile(optimizer=keras.optimizers.Adam())

vae.fit(x=x_train, y=None, 
        epochs=epochs, 
        batch_size=batch_size, 
        verbose=False,
        validation_data=(x_test, None))

Я хочу оценить модель на отдельных наборах данных после обучения (поскольку у меня есть несколько наборов данных для оценки, я не использую ее как validation_data). Однако, когда я пытаюсь бежатьvae.evaluate(data), он возвращает пустой список [].

Примечание. Я могу легко получить показатели обучения и проверки с помощью vae.history.history, но проблема в том, когда я пытаюсь оценить после тренировки. Но когда я пытаюсь вернуть метрикиvae.metricsон также возвращает пустой список. Как я могу получитьmodel.evaluateработать с настраиваемой процедурой обучения, которая возвращает данные о потерях? Нужно ли мне определять что-то особенное для оценки?

Вот как определяется класс VAE. Более подробную информацию можно найти в руководстве выше.

class VAE(keras.Model):
    def __init__(self, encoder, decoder, **kwargs):
        super(VAE, self).__init__(**kwargs)
        self.encoder = encoder
        self.decoder = decoder

    def train_step(self, data):
        if isinstance(data, tuple):
            data = data[0]
        with tf.GradientTape() as tape:
            z_mean, z_log_var, z = encoder(data)
            reconstruction = decoder(z)
            reconstruction_loss = tf.reduce_mean(
                keras.losses.binary_crossentropy(data, reconstruction)
            )
            reconstruction_loss *= 28 * 28
            kl_loss = 1 + z_log_var - tf.square(z_mean) - tf.exp(z_log_var)
            kl_loss = tf.reduce_mean(kl_loss)
            kl_loss *= -0.5
            total_loss = reconstruction_loss + kl_loss
        grads = tape.gradient(total_loss, self.trainable_weights)
        self.optimizer.apply_gradients(zip(grads, self.trainable_weights))
        return {
            "loss": total_loss,
            "reconstruction_loss": reconstruction_loss,
            "kl_loss": kl_loss,
        }