From 903899c66faccafb919933c21014e694de97d93d Mon Sep 17 00:00:00 2001
From: Ilya Sudakov <108363692+sudakovcom@users.noreply.github.com>
Date: Sat, 8 Feb 2025 18:51:53 +0300
Subject: [PATCH] Update 02_hw_StarGAN.ipynb

lpips computes score for a batch of data (produces nonconsistent assert as batch_size is specified by user)
---
 Homeworks/02_hw_StarGAN.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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"