diff --git a/Homeworks/02_hw_StarGAN.ipynb b/Homeworks/02_hw_StarGAN.ipynb index c29c4569..326d7f4a 100644 --- a/Homeworks/02_hw_StarGAN.ipynb +++ b/Homeworks/02_hw_StarGAN.ipynb @@ -235,7 +235,7 @@ " raise NotImplementedError(\"Реализуйте генерацию изображения x_fake с использованием вашей модели\")\n", " \n", " # Вычисление LPIPS между сгенерированным и реальным изображением\n", - " values.append(lpips(x_fake.cpu(), x_real.cpu()).squeeze().item())\n", + " values.append(lpips(x_fake.cpu(), x_real.cpu()).squeeze().item()/batch_size)\n", "\n", "print(\"Среднее значение LPIPS:\", np.mean(values))\n", "assert np.mean(values) < 1.3"