pytorch/examples

Gradient vanishing of G in the DCGAN example

Open

#822 opened on 2020年9月11日

GitHub で見る
 (0 comments) (0 reactions) (0 assignees)Python (21,634 stars) (9,429 forks)batch import
help wanted

説明

Hello,

I have trained the DCGAN with the default hyper-parameter settings on the downloaded "img_align_celeba" dataset (recommended in the tutorial). However, the results reveal strong gradient vanishing of G. While Loss_D keeps decreasing towards 0, Loss_G grows high (towards 100).

It seems that D is trained so well, preventing a good training on G. I didn't do any modifications on the code. Do you know what happened?

Thanks!

コントリビューターガイド