WGAN-GP создает черные изображения для набора данных MNIST - PullRequest
0 голосов
/ 28 февраля 2019

Я использую следующую реализацию WGAN-GP из keras-contrib repo без каких-либо изменений: https://github.com/keras-team/keras-contrib/blob/master/examples/improved_wgan.py

Код по умолчанию работает на наборе данных MNIST, но приводит к черным образцам, полученным генератором после 10эпохи и образцы остаются черными в предстоящие эпохи и никогда не восстанавливаются (потеря критика также перестает уменьшаться).Ниже приведены образцы в разные эпохи:

Эпоха 0: enter image description here

Эпоха 3: enter image description here

Эпоха 6: enter image description here

Эпоха 10: enter image description here

Эпоха 41: enter image description here

Не могу разобраться с проблемой здесь.Пожалуйста, помогите.

...