Installation Instructions

python version 3.8
pip install -r requirements.txt
cd gym
pip install -e .
cd ../nflows
pip install -e .
cd ..

Run 1-D Envs

python -u Uncertain_1d.py \
        --model nflows_ensemble 
        --num_layers 1 \
        --hids  20 \
        --domain 1.2 \
        --env hetero \
        --acquisition_type kl_exp \
        --test_acquisition \
        --epochs 6000 \
        --dropout_masks \
        --base_distro \
        --ensemble_size 5
