This code is forked from the official code by Sendera et al. (2024).

Example command to compare random and uniform 10-step discretizations (TB without Langevin parametrization on Manywell):

```
python train.py --energy many_well --t_scale 1.0 --pis_architectures --zero_init --clipping --discretizer_traj_length 10 --mode_fwd tb --lr_policy 1e-3 --lr_flow 1e-1 --discretizer random --seed 1
python train.py --energy many_well --t_scale 1.0 --pis_architectures --zero_init --clipping --discretizer_traj_length 10 --mode_fwd tb --lr_policy 1e-3 --lr_flow 1e-1 --discretizer uniform --seed 1
```