fix min snr logic, thanks to @justinlovelace

Author: lucidrains

rin_pytorch/rin_pytorch.py

@@ -836,7 +836,7 @@ def forward(self, img, *args, **kwargs):
         maybe_clipped_snr = snr.clone()
 
         if self.min_snr_loss_weight:
-            maybe_clipped_snr.clamp_(min = self.min_snr_gamma)
+            maybe_clipped_snr.clamp_(max = self.min_snr_gamma)
 
         if self.objective == 'eps':
             loss_weight = maybe_clipped_snr / snr

setup.py

@@ -3,7 +3,7 @@
 setup(
   name = 'RIN-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.7.1',
+  version = '0.7.2',
   license='MIT',
   description = 'RIN - Recurrent Interface Network - Pytorch',
   author = 'Phil Wang',