wrong discriminator is freezed

[Kanbe-Kotori]

In L468 of gan_aug_cifar_sm.py, the model that should be frozen is discriminator_aug, while the one that is actually frozen is discriminator. Obviously, the discriminator does not appear below, while discriminator_aug not only appears, but also be updated by g_loss, so I would like to know whether the "Freeze D" mentioned in the proposed algorithm is implemented correctly?

This problem seems also occur in the code of GANs corresponding to other datasets.

[conditionWang]

Thanks for pointing this out. We just uploaded the old version of unfreezing the discriminator. We have updated the code.