Two types of train_step in tensorflow2?

Question

I\'m training a GAN and I\'d like to train the discriminator more at first (first 5 epochs, for example), and then change it up and train both the discriminator and generator eq