VITA-Group / EnlightenGAN

[IEEE TIP] "EnlightenGAN: Deep Light Enhancement without Paired Supervision" by Yifan Jiang, Xinyu Gong, Ding Liu, Yu Cheng, Chen Fang, Xiaohui Shen, Jianchao Yang, Pan Zhou, Zhangyang Wang
Other
890 stars 198 forks source link

where is the loss in your code #41

Closed lyyyyynnnnnn closed 4 years ago

TrinhQuocNguyen commented 4 years ago

I think it's in this function.

single_model.py file:

    def get_current_errors(self, epoch):
        D_A = self.loss_D_A.data
        D_P = self.loss_D_P.data if self.opt.patchD else 0
        G_A = self.loss_G_A.data
        if self.opt.vgg > 0:
            vgg = self.loss_vgg_b.data/self.opt.vgg if self.opt.vgg > 0 else 0
            return OrderedDict([('D_A', D_A), ('G_A', G_A), ("vgg", vgg), ("D_P", D_P)])
        elif self.opt.fcn > 0:
            fcn = self.loss_fcn_b.data/self.opt.fcn if self.opt.fcn > 0 else 0
            return OrderedDict([('D_A', D_A), ('G_A', G_A), ("fcn", fcn), ("D_P", D_P)])