Skip to content

Commit

Permalink
Use more numerically stable log1p in examples/scanvi/scanvi.py (#3397)
Browse files Browse the repository at this point in the history
  • Loading branch information
kit1980 authored Sep 18, 2024
1 parent c8b6020 commit e914e19
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/scanvi/scanvi.py
Original file line number Diff line number Diff line change
Expand Up @@ -104,7 +104,7 @@ def forward(self, x):
# Transform the counts x to log space for increased numerical stability.
# Note that we only use this transform here; in particular the observation
# distribution in the model is a proper count distribution.
x = torch.log(1 + x)
x = torch.log1p(x)
h1, h2 = split_in_half(self.fc(x))
z2_loc, z2_scale = h1[..., :-1], softplus(h2[..., :-1])
l_loc, l_scale = h1[..., -1:], softplus(h2[..., -1:])
Expand Down

0 comments on commit e914e19

Please sign in to comment.