fix: sigmoid norm

archinetai · Oct 5, 2022 · d32c307 · d32c307
1 parent 13cddd6
commit d32c307
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/audio_diffusion_pytorch/utils.py b/audio_diffusion_pytorch/utils.py
@@ -87,14 +87,14 @@ def upsample(waveforms: Tensor, factor: int, **kwargs) -> Tensor:
 
 def wave_norm(x: Tensor, peak: float = 0.5, alpha: float = 20.0) -> Tensor:
     x = x.clip(-1, 1)
-    x = torch.sigmoid(alpha * x)
+    x = 2 * torch.sigmoid(alpha * x) - 1
     x = x.clip(-1, 1)
     return x * peak
 
 
 def wave_unnorm(x: Tensor, peak: float = 0.5, alpha: float = 20.0) -> Tensor:
     x = x / peak
     x = x.clip(-1, 1)
-    x = (1.0 / alpha) * torch.log(x / (1 - x))
+    x = (1.0 / alpha) * torch.log((x + 1) / (1 - x))
     x = x.clip(-1, 1)
     return x
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
     name="audio-diffusion-pytorch",
     packages=find_packages(exclude=[]),
-    version="0.0.55",
+    version="0.0.54",
     license="MIT",
     description="Audio Diffusion - PyTorch",
     long_description_content_type="text/markdown",