You can install python dependencies using conda.
condaa create -y -f environment.yaml
CONDA_OVERRIDE_CUDA=11.7 mamba create -y -f environment.yaml
setup.zsh
exports two environment variables named PROJECT_DATA_DIR
and PROJECT_LOG_DIR
.
PROJECT_DATA_DIR
is a directory where datasets are stored and PROJECT_LOG_DIR
is a directory where training output files are saved.
Edit them to suit your system.
source setup.zsh
You can download ttbar samples using the following script.
./scripts/download-files.zsh
All configuration is handled using yaml file.
python train.py ./config/transformer.yaml