Method optimize_parameters

models/hdnet_model.py:59–64

(self)

        self.loss_G.backward()

    def optimize_parameters(self):
        self.forward()
         # update G
        self.optimizer_G.zero_grad()  # set G's gradients to zero
        self.backward_G()  # calculate graidents for G
        self.optimizer_G.step()  # udpate G's weights