Spaces:

ucalyptus
/

PTI

Runtime error

PTI / criteria /l2_loss.py

simp

2d7efb8 over 2 years ago

No virus

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