Предположим, у меня есть простая модель CNN с двумя слоями Conv2D, я обучил эту модель на своем наборе данных изображений, я собираюсь передать параметры этой модели CNN в VAE (в качестве входных данных кодировщика), чтобы сначала уменьшить их параметры в пространство вложения (Z или скрытое пространство VAE). Затем я хочу восстановить параметры CNN (с их исходными размерами), используя выходные данные декодера VAE.
Я не знаю, как я могу реализовать это в PyTorch и как бы мне это сделать. восстановить вектор параметров до параметров модели CNN.
Заранее спасибо!
Подробнее здесь: https://stackoverflow.com/questions/784 ... sing-a-vae
