rikyaa

I think there is a wrong!

5

# Training Stage with Concealing Field if mode != 'test': if i_level == 0: accum_prod_dark = torch.cat([torch.ones_like(alpha[..., :1] * coarse_dark[0]), torch.cumprod(1.0 - alpha[..., :-1] + eps, dim=-1) * coarse_dark[1:]], dim=-1,)...

您好，我想把psnr，ssim等数据输出到console,能否提供代码思路，谢谢您！