diff --git a/README.md b/README.md index 187a7be..064c9c9 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,5 @@ + + # eCLIP @@ -13,6 +15,8 @@ Read the full paper here: https://arxiv.org/abs/2403.10153 ### Setup +The repository requires Python >= 3.11. + We use the `eclip` package for training and evaluation. To install it run `pip install -e .` from the root folder and it installs all the dependencies. We use `pytorch-lightning` to train the model and `hydra` to handle the config files. @@ -65,4 +69,3 @@ We use `Mistral 7B Instruction` for generating radiology reports using the input - Tracking -- [Weights & Biases](https://wandb.ai/) - Config Management -- [Hydra](https://hydra.cc/) - Data loading -- [Webdataset](https://github.com/webdataset/webdataset) -