fix mini_batch_size

arrjon · arrjon · commit 983cb8d39909 · 2025-09-26T18:16:02.000+02:00
diff --git a/bayesflow/networks/diffusion_model/compositional_diffusion_model.py b/bayesflow/networks/diffusion_model/compositional_diffusion_model.py
@@ -310,7 +310,7 @@ def _inverse_compositional(
                     "or use another backend."
                 )
         else:
-            mini_batch_size = min(integrate_kwargs.pop("mini_batch_size", int(n_compositional * 0.1)), 1)
+            mini_batch_size = max(integrate_kwargs.pop("mini_batch_size", int(n_compositional * 0.1)), 1)
         self.compositional_bridge_d0 = float(integrate_kwargs.pop("compositional_bridge_d0", 1.0))
         self.compositional_bridge_d1 = float(integrate_kwargs.pop("compositional_bridge_d1", 1.0))
 

Original file line number	Diff line number	Diff line change
`@@ -310,7 +310,7 @@ def _inverse_compositional(`
`310`	`310`	`"or use another backend."`
`311`	`311`	`)`
`312`	`312`	`else:`
`313`		`- mini_batch_size = min(integrate_kwargs.pop("mini_batch_size", int(n_compositional * 0.1)), 1)`
	`313`	`+ mini_batch_size = max(integrate_kwargs.pop("mini_batch_size", int(n_compositional * 0.1)), 1)`
`314`	`314`	`self.compositional_bridge_d0 = float(integrate_kwargs.pop("compositional_bridge_d0", 1.0))`
`315`	`315`	`self.compositional_bridge_d1 = float(integrate_kwargs.pop("compositional_bridge_d1", 1.0))`
`316`	`316`