hotfix

Vermeille · Sep 4, 2024 · 0ebc106 · 0ebc106
1 parent 5c6c3da
commit 0ebc106
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchelie/nn/layers.py b/torchelie/nn/layers.py
@@ -224,7 +224,7 @@ def forward(self, x):
             out = tchf.unit_gaussian_prior(mu, sigma, strength)
             return out if not self.return_std else (out, sigma)
         else:
-            return mu if not self.return_std else (mu, sigma
+            return mu if not self.return_std else (mu, sigma)
 
 
 class InformationBottleneck(UnitGaussianPrior):