First, install required packages.
cd rla_simulation
pip3 install -r requirements.txt
Second, set up the environment
source env.sh
Download the data root file -- for instance to data folder after creating it -- and then use the following command to run a training.
mkdir data
cd data
wget https://example.com/example/training_LARGE.root
cd ..
Adjust configs/vae_1.yaml
file accordingly and then run the training as:
python3 rlasim/bin/train_conditional_vae.py
And then run the test with:
python3 rlasim/bin/train_conditional_vae.py --predict