22 lines
504 B (Stored with Git LFS)
YAML
22 lines
504 B (Stored with Git LFS)
YAML
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]
|