fc8951bbd36b14099e41fe171ab57e9f3937fe4f,congan_train.py,,train,#,191

Before Change

            end = timer(); print(f"---load real imgs elapsed time: {end-start}")

            start = timer()
            real_data = autograd.Variable(r_image.cuda(),requires_grad=True)
            //TODO: check load labels
            real_label = autograd.Variable(r_label.cuda())

After Change

        print(f"---train G elapsed time: {end - start}")
        //---------------------TRAIN D------------------------
        for p in aD.parameters():  // reset requires_grad
            p.requires_grad_(True)  // they are set to False below in training G
        for i in range(CRITIC_ITERS):
            print("Critic iter: " + str(i))
            

Instances