Solved by Sayed_Nadim in post #2 
              
              
                Hi, 
Can you train the generator first and then the discriminator? The discriminator seems to detach the output from graph for loss calculations.