PTI / criteria /l2_loss.py
ucalyptus's picture
simp
2d7efb8
raw
history blame
181 Bytes
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