First, install required packages.
cd rla_simulation
pip3 install -r requirements.txtSecond, set up the environment
source env.shDownload 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.pyAnd then run the test with:
python3 rlasim/bin/train_conditional_vae.py --predict