diff --git a/mnist_gan.py b/mnist_gan.py index 3506a8c..3c2aa05 100644 --- a/mnist_gan.py +++ b/mnist_gan.py @@ -100,10 +100,6 @@ def make_trainable(net, val): discriminator.summary() # Freeze weights in the discriminator for stacked training -def make_trainable(net, val): - net.trainable = val - for l in net.layers: - l.trainable = val make_trainable(discriminator, False) # Build stacked GAN model