🩹 Fix outdated example

francois-rozet · Oct 30, 2023 · d934345 · d934345
1 parent 68dfac0
commit d934345
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -50,9 +50,8 @@ Like `torch.nn` built-in components, these classes are based on functional defin
 from piqa.ssim import ssim
 from piqa.utils.functional import gaussian_kernel
 
-kernel = gaussian_kernel(11, sigma=1.5).expand(3, 11, 11)
-
-l = 1 - ssim(x, y, kernel=kernel)
+kernel = gaussian_kernel(11, sigma=1.5).repeat(3, 1, 1)
+ss, cs = ssim(x, y, kernel=kernel)
 ```
 
 For more information, check out the documentation at [piqa.readthedocs.io](https://piqa.readthedocs.io).

diff --git a/docs/index.rst b/docs/index.rst
@@ -56,9 +56,8 @@ Like :mod:`torch.nn` built-in components, these classes are based on functional
     from piqa.ssim import ssim
     from piqa.utils.functional import gaussian_kernel
 
-    kernel = gaussian_kernel(11, sigma=1.5).expand(3, 11, 11)
-
-    l = 1 - ssim(x, y, kernel=kernel)
+    kernel = gaussian_kernel(11, sigma=1.5).repeat(3, 1, 1)
+    ss, cs = ssim(x, y, kernel=kernel)
 
 For more information, check out the `API <api.html>`_ or the `repository <https://github.com/francois-rozet/piqa>`_.