Backport PR #1445: Remove softplus in AmortizedLDA for cell topic var…

…iance
scverse · Mar 20, 2022 · ba22ce2 · ba22ce2
1 parent 54b0560
commit ba22ce2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scvi/module/_amortizedlda.py b/scvi/module/_amortizedlda.py
@@ -204,7 +204,7 @@ def forward(
             pyro.sample(
                 "log_cell_topic_dist",
                 dist.Normal(
-                    cell_topic_posterior_mu, F.softplus(cell_topic_posterior_sigma)
+                    cell_topic_posterior_mu, cell_topic_posterior_sigma
                 ).to_event(1),
             )