task:
  benchmark_suite: sbibm
  name: sir
  num_posterior_samples: 10000
training:
  initial_learning_rate: 5e-4
  optimizer: adamw
  mode: offline
  num_simulations: 100000
  epochs: 100
  batch_size: 64
architecture:
  summary_network:
    type: "bayesflow.networks>TimeSeriesNetwork"
  inference_network:
    type: "bayesflow.networks>FreeFormFlow"
    kwargs:
      beta: 25.0
      encoder_subnet_kwargs:
        widths: [128, 128, 128]
      decoder_subnet_kwargs:
        widths: [128, 128, 128]