what's real critertion in main.py?

[zhizhengyan]

Dear Sir,
    I find some contradictions in your code about main.py file that confused me. Thus, I wanna to verify this question.
In L125, you declare critertion=nn.L1Loss(),

But according to psnr formula, we must get mse value that is L2Loss(MSELoss). Your code L87 using L1Loss to compute mse, is it mae?

    looking foward to hearing from you interpretation. Thanks.