task: benchmark_suite: sbibm name: two_moons num_posterior_samples: 10000 training: initial_learning_rate: 5e-4 optimizer: adamw mode: offline num_simulations: 100000 epochs: 100 batch_size: 32 architecture: summary_network: inference_network: type: "bayesflow.networks>DiffusionModel" kwargs: