dolche

Когда обучаем сеть , чтобы выводить на экран процесс обучения и изменение картинок, делаем это через функцию с эпох

# Функция, которую будем запускать после каждой эпохи
def on_epoch_end(epoch, logs):
    if epoch in save_epochs:
        clear_output() 
        # Сравнение реальных и декодированных цифр
        decoded = vae.predict()
        plot_img(#функция для выведения изображений)</code

типа такого