Skip to content

Commit b034b21

Browse files
committed
Adapt readme for speed and performance
1 parent 87a977d commit b034b21

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,14 +37,14 @@ Using the high-level interface is easy, as demonstrated by the minimal working e
3737
import bayesflow as bf
3838

3939
workflow = bf.BasicWorkflow(
40-
inference_network=bf.networks.FlowMatching(),
41-
summary_network=bf.networks.TimeSeriesTransformer(),
40+
inference_network=bf.networks.CouplingFlow(),
41+
summary_network=bf.networks.TimeSeriesNetwork(),
4242
inference_variables=["parameters"],
4343
summary_variables=["observables"],
4444
simulator=bf.simulators.SIR()
4545
)
4646

47-
history = workflow.fit_online(epochs=50, batch_size=32, num_batches_per_epoch=500)
47+
history = workflow.fit_online(epochs=15, batch_size=32, num_batches_per_epoch=200)
4848

4949
diagnostics = workflow.plot_default_diagnostics(test_data=300)
5050
```

0 commit comments

Comments
 (0)