To use the repo create a conda env using the following command
conda create --name cfreg python=3.11
Once the env has been created use:
conda activate cfreg
Then install the required libreries:
python -m pip install matplotlib wandb hydra-core scikit-learn lightning torch pytorch-lightning omegaconf