## Installation

1. Install torch: `pip install torch==2.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118`
1. Install packages: `pip install submitit hydra-core hydra-submitit-launcher loguru tqdm pytest kymatio  pytorch-lightning timm lightly einops GitPython wandb medmnist`


## Tests

To run tests `python -m pytest tests/`
