Open ChenYutongTHU opened 4 years ago
Thanks for this awesome project.
I note that during training inpainter, L1_loss is measured on the pair of complete output img and complete input image.
gen_l1_loss = self.l1_loss(outputs, images) * self.config.L1_LOSS_WEIGHT / torch.mean(masks) gen_loss += gen_l1_loss
I wonder if it would be more proper to only compare the error only within the masked region since the final output image is a merged one of generator's restored masked region and the given context.
Thanks!
Thanks for this awesome project.
I note that during training inpainter, L1_loss is measured on the pair of complete output img and complete input image.
gen_l1_loss = self.l1_loss(outputs, images) * self.config.L1_LOSS_WEIGHT / torch.mean(masks) gen_loss += gen_l1_loss
I wonder if it would be more proper to only compare the error only within the masked region since the final output image is a merged one of generator's restored masked region and the given context.
Thanks!