File size: 181 Bytes
822c5d5
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
import torch

l2_criterion = torch.nn.MSELoss(reduction='mean')


def l2_loss(real_images, generated_images):
    loss = l2_criterion(real_images, generated_images)
    return loss