Spaces:
Runtime error
Runtime error
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
|