Функция потерь StyleGAN почти равна 0 в черно-белых изображениях ⇐ Python

[Anonymous]

Я обучаю StyleGAN для проекта, в котором хочу создавать черно-белые изображения для увеличения данных. У меня недостаточно знаний о StyleGAN, поэтому я ищу пример в Интернете в PyTorch.
Я нашел этот: https://www.kaggle.com/code/tauilabdelilah/stylegan -implementation-from-scratch-pytorch/notebook
Я провел несколько тестов с кодом, и кажется, что он генерирует хорошие изображения. Проблема в том, что график функции потерь кажется странным. Потери дискриминатора и генератора достигают почти 0 в определенную эпоху и остаются стабильными на этом значении.
[img]https://i.sstatic .net/LS5QL6dr.png[/img]

Я хотел бы знать, является ли это нормальной функцией потерь StyleGAN (я не нашел ни одного графика StyleGAN) и так ли это нет, возможно, кто-то мог бы помочь мне улучшить потери или штраф за градиент в коде.
Изображения одноканальные (очевидно) и 128x128; Я не модифицировал код с точки зрения обучения моделей или штрафа за градиент, поэтому вы можете посмотреть код здесь: https://www.kaggle.com/code/tauilabdeli ... tion-from-
Большое спасибо.

Подробнее здесь: https://stackoverflow.com/questions/791 ... -bw-images