nurion4hep / KSC-RnD-2024-Transformer4MET

0 stars 4 forks source link

Differential MET Reconstruction

Updates at Here!

Google Docs

Recipes

Install

Micromamba

"${SHELL}" <(curl -L micro.mamba.pm/install.sh)

copied from Micromamba Installation

Python environment

You can install python packages with environment.yaml

micromamba create -y -f environment.yaml

Setup

let's start setup!

source setup.sh

Run a training code on a local machine

You can then train a deep learning model with train.py and a configuration .yaml file. For example, you can a Transformer-based network that takes in L1 ParticleFlow candidates and reconstruct MET.

python train.py -c ./config/test-l1pf-transformer-neuron.yaml

If you want to understand how to configure it, I recommend reading Configure hyperparameters from the CLI

Submit a batch job with HTCondor

Before submitting a job, u need to check available memory.

condor_status -compact -constraint 'Gpus > 0'

You can submit a training job to a GPU cluster using submit.py

submit.py --config ./config/only-high-met.yaml --data ./config/data/delphes/knu/sanity-check.yaml --trainer ./config/trainer/sanity-check.yaml -e sanity-check -m 50GB
condor_q $USER