diff --git a/piq/__init__.py b/piq/__init__.py index 899effb0..9676c55e 100644 --- a/piq/__init__.py +++ b/piq/__init__.py @@ -1,4 +1,4 @@ -__version__ = "0.5.3" +__version__ = "0.5.4" from .ssim import ssim, multi_scale_ssim, SSIMLoss, MultiScaleSSIMLoss from .msid import MSID diff --git a/piq/pieapp.py b/piq/pieapp.py index 86d274f0..e373954d 100644 --- a/piq/pieapp.py +++ b/piq/pieapp.py @@ -119,7 +119,7 @@ class PieAPP(_Loss): .. [2] https://github.com/prashnani/PerceptualImageError """ - _weights_url = "https://github.com/photosynthesis-team/piq/releases/download/v0.5.3/PieAPPv0.1.pth" + _weights_url = "https://github.com/photosynthesis-team/piq/releases/download/v0.5.4/PieAPPv0.1.pth" def __init__( self,