Skip to content

Commit

Permalink
Use Cauchy prior fist for bias
Browse files Browse the repository at this point in the history
  • Loading branch information
ziatdinovmax committed Feb 26, 2024
1 parent 1af7c07 commit d82b1b9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion gpax/models/dkl.py
Original file line number Diff line number Diff line change
Expand Up @@ -159,7 +159,7 @@ def sample_weights(name: str, in_channels: int, out_channels: int) -> jnp.ndarra

def sample_biases(name: str, channels: int) -> jnp.ndarray:
"""Sampling bias vector"""
b = numpyro.sample(name=name, fn=dist.Normal(
b = numpyro.sample(name=name, fn=dist.Cauchy(
loc=jnp.zeros((channels)), scale=jnp.ones((channels))))
return b

Expand Down

0 comments on commit d82b1b9

Please sign in to comment.