/rla_simulation

In works.

Primary LanguagePython

Detector simulation with generative models

Quick start

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