Имплементирование процедуры обучения WGAN

Необходимо имплементировать процедуру обучения WGAN. В качестве датасета предлагается использовать fashion mnist, но вы можете выбрать другой. Необходимые моменты (скачивание датасета, базовая модель GAN, базовый цикл обучения и др.) реализованы в ноутбке, который предлагается вам как старт вашей работы. В нем же вы найдете полезные ссылки для решения задачи.

В качестве шаблона решения используйте ноутбук gan_hw.ipynb (лежит в архиве gan_hw). Реализуйте обучение WGAN с клиппингом весов и с gradient penalty и условную генерацию. Опишите полученные результаты, визуализируйте процедуру обучения в виде графиков и результаты в виде изображений.

Необходимо реализовать WGAN-GP (со штрафом на градиенты) и WGAN (с клиппингом весов).

Желательно реализовать еще условную генерацию (можете использовать с любым типом GAN).

Подробности в ноутбуке gan_hw.ipynb (лежит в архиве gan_hw).