Skip to content

Commit 1129b1c

Browse files
committed
merge new samplers
1 parent a661204 commit 1129b1c

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

bayesflow/networks/diffusion_model/compositional_diffusion_model.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,14 +32,13 @@ class CompositionalDiffusionModel(DiffusionModel):
3232
"activation": "mish",
3333
"kernel_initializer": "he_normal",
3434
"residual": True,
35-
"dropout": 0.0,
35+
"dropout": 0.05,
3636
"spectral_normalization": False,
3737
}
3838

3939
INTEGRATE_DEFAULT_CONFIG = {
40-
"method": "euler_maruyama",
41-
"corrector_steps": 1,
42-
"steps": 100,
40+
"method": "two_step_adaptive",
41+
"steps": "adaptive",
4342
}
4443

4544
def __init__(

0 commit comments

Comments
 (0)