fix: wrong vae bottleneck

archinetai · Nov 7, 2022 · 2ca7120 · 2ca7120
1 parent 2410c7a
commit 2ca7120
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/audio_encoders_pytorch/modules.py b/audio_encoders_pytorch/modules.py
@@ -581,8 +581,8 @@ def forward(
     ) -> Union[Tensor, Tuple[Tensor, Any]]:
         mean_and_std = self.to_mean_and_std(x)
         mean, std = mean_and_std.chunk(chunks=2, dim=1)
-        std = torch.tanh(std)  # mean in range [-1, 1]
-        mean = torch.tanh(mean) + 1.0  # std in range [0, 2]
+        mean = torch.tanh(mean)  # mean in range [-1, 1]
+        std = torch.tanh(std) + 1.0  # std in range [0, 2]
         out = gaussian_sample(mean, std)
         info = dict(
             variational_kl_loss=kl_loss(mean, std) * self.loss_weight,

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
     name="audio-encoders-pytorch",
     packages=find_packages(exclude=[]),
-    version="0.0.12",
+    version="0.0.13",
     license="MIT",
     description="Audio Encoders - PyTorch",
     long_description_content_type="text/markdown",