Skip to content

Commit 29e082a

Browse files
committed
Include baselines in README
1 parent b97b75f commit 29e082a

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

README.md

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,16 +31,15 @@ to install the dependencies and setup a virtual environment. Either activate the
3131
The best way to get started is to look at [the jupyter notebook](notebooks/tps_gaussian.ipynb) which contains the necessary code for 2D potentials in one place.
3232

3333
## Running the baselines
34-
35-
To run the baselines (i.e., TPS with shooting) you can run
34+
You can either use the TPS shooting baselines [provided by us](https://github.com/plainerman/variational-doob/releases/tag/camera-ready), or re-create them by running
3635

3736
```bash
3837
python tps_baseline_mueller.py
3938
# For this to work, you need to specify which baselines to run by changing the all_paths variable in the script
4039
python eval/evaluate_mueller.py
4140
```
4241

43-
for a toy-potential and
42+
to generate and evaluate transitions for the Müller-Brown toy-potential or use
4443

4544
```bash
4645
python tps_baseline.py --mechanism two-way-shooting --num_paths 1000 --states phi-psi
@@ -50,7 +49,9 @@ python tps_baseline.py --mechanism two-way-shooting --num_paths 1000 --states rm
5049
python eval/evaluate_tps.py
5150
```
5251

53-
for ALDP respectively. In both cases, you might need to change the paths that you want to evaluate.
52+
for ALDP respectively.
53+
54+
**Note:** In both cases, you might need to change the paths that you want to evaluate in ``evaluate_mueller.py` or `evaluate_tps.py`.
5455

5556
## Run our method
5657
To sample trajectories with our method, we provide ready to go config files in `configs/`. You can run them with

0 commit comments

Comments
 (0)