torch-l1-snr 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

torch_l1_snr/__init__.py

@@ -14,4 +14,4 @@ __all__ = [
     "MultiL1SNRDBLoss",
 ] 
 
-__version__ = "0.1.1"
+__version__ = "0.1.2"

torch_l1_snr/l1snr.py

@@ -104,9 +104,10 @@ class L1SNRLoss(torch.nn.Module):
         l1snr_loss = torch.mean(d1)
 
         c = 10.0 / math.log(10.0)
-        inv_mean = torch.mean(1.0 / (l1_error.detach() + self.eps))
-        # w-independent scaling to match typical gradient magnitudes
-        scale_time = c * inv_mean
+        # Scale by reference signal magnitude (not error) to preserve gradient distinction
+        # L1SNR has inverse-error gradients; L1 should have uniform gradients
+        inv_ref_mean = torch.mean(1.0 / (l1_true.detach() + self.eps))
+        scale_time = c * inv_ref_mean
         l1_term = torch.mean(l1_error) * scale_time
 
         loss = (1.0 - w) * l1snr_loss + w * l1_term
@@ -446,10 +447,11 @@ class STFTL1SNRDBLoss(torch.nn.Module):
         w = float(self.l1_weight)
         if 0.0 < w < 1.0:
             c = 10.0 / math.log(10.0)
-            inv_mean_comp = torch.mean(0.5 * (1.0 / (err_re.detach() + self.l1snr_eps) +
-                                              1.0 / (err_im.detach() + self.l1snr_eps)))
-            # w-independent scaling to match typical gradient magnitudes (factor 2.0 for Re/Im symmetry)
-            scale_spec = 2.0 * c * inv_mean_comp
+            # Scale by reference signal magnitude (not error) to preserve gradient distinction
+            # L1SNR has inverse-error gradients; L1 should have uniform gradients
+            inv_ref_mean_comp = torch.mean(0.5 * (1.0 / (ref_re.detach() + self.l1snr_eps) +
+                                                  1.0 / (ref_im.detach() + self.l1snr_eps)))
+            scale_spec = 2.0 * c * inv_ref_mean_comp
             l1_term = 0.5 * (torch.mean(err_re) + torch.mean(err_im)) * scale_spec
 
             loss = (1.0 - w) * d1_sum + w * l1_term

{torch_l1_snr-0.1.1.dist-info → torch_l1_snr-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: torch-l1-snr
-Version: 0.1.1
+Version: 0.1.2
 Summary: L1-SNR loss functions for audio source separation in PyTorch
 Home-page: https://github.com/crlandsc/torch-l1-snr
 Author: Christopher Landschoot
@@ -237,7 +237,7 @@ While this can potentially reduce the "cleanliness" of separations and slightly
 
 The implementation is optimized for efficiency: if `l1_weight` is `0.0` or `1.0`, the unused loss component is not computed, saving computational resources.
 
-**Note on Gradient Balancing:** When blending losses (`0.0 < l1_weight < 1.0`), the implementation automatically scales the L1 component to approximately match the gradient magnitudes of the L1SNR component. This helps maintain stable training without manual tuning.
+**Note on Gradient Balancing:** When blending losses (`0.0 < l1_weight < 1.0`), the implementation automatically scales the L1 component to approximately match gradient magnitudes while preserving distinct gradient behaviors. This helps maintain stable training without manual tuning.
 
 ## Limitations
 

torch_l1_snr-0.1.2.dist-info/RECORD

@@ -0,0 +1,7 @@
+torch_l1_snr/__init__.py,sha256=eJvtoXxJqrrayRFH1xsx9oYMC-aobnh5VK1SaxjsytU,244
+torch_l1_snr/l1snr.py,sha256=aA3BlPZRNsXGAHWnk3PFY8lDiH4HcWeFuhmM4gAKYiQ,34711
+torch_l1_snr-0.1.2.dist-info/licenses/LICENSE,sha256=JdS2Pv6DDs3jvXHACGdcHYdiFMe9EO1XGeHkEHLTr8Y,1079
+torch_l1_snr-0.1.2.dist-info/METADATA,sha256=hBBGDWtSD31aCu31K6bMZuTRBCqCx5UYg6WPXrxsBn4,15722
+torch_l1_snr-0.1.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+torch_l1_snr-0.1.2.dist-info/top_level.txt,sha256=VUo0QlGvu7tOF8BKWWDoIiLlhcAcetYwR6c8Ldhhpco,13
+torch_l1_snr-0.1.2.dist-info/RECORD,,

torch_l1_snr-0.1.1.dist-info/RECORD

@@ -1,7 +0,0 @@
-torch_l1_snr/__init__.py,sha256=L3Cpdpnhz80gpfcTf6aNM1ROPdOIbdNoN8vO9LxcZEQ,244
-torch_l1_snr/l1snr.py,sha256=F1NF3VGodaLWFtHs9xco9MbxfEJ01ip_JSHFS2GgBkU,34520
-torch_l1_snr-0.1.1.dist-info/licenses/LICENSE,sha256=JdS2Pv6DDs3jvXHACGdcHYdiFMe9EO1XGeHkEHLTr8Y,1079
-torch_l1_snr-0.1.1.dist-info/METADATA,sha256=8zC2S_NgV8B4Wg59QJwoupNfF063TxD6nFEzXJdeZIw,15704
-torch_l1_snr-0.1.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-torch_l1_snr-0.1.1.dist-info/top_level.txt,sha256=VUo0QlGvu7tOF8BKWWDoIiLlhcAcetYwR6c8Ldhhpco,13
-torch_l1_snr-0.1.1.dist-info/RECORD,,