Update README.md

README.md

@@ -5,10 +5,11 @@ license: cc-by-4.0
 # LoRA-Ensemble: Uncertainty Modelling for Self-attention Networks
 Michelle Halbheer, Dominik J. Mühlematter, Alexander Becker, Dominik Narnhofer, Helge Aasen, Konrad Schindler and Mehmet Ozgur Turkoglu - 2024
 
-## Pretrained models
-This repository contains the pretrained models corresponding to the code we released on [GitHub](https://github.com/prs-eth/LoRA-Ensemble/).
-The usage of the models with our pipeline is described on GitHub.
-This repository only contains the models for our final experiments per dataset, not, however, for all intermediate results.
+## Pre-trained models
+This repository contains the pre-trained models corresponding to the code we released on [GitHub](https://github.com/prs-eth/LoRA-Ensemble/).
+The usage of the models with our pipeline is described in the GitHub repository.
+This repository only contains the models for our final experiments on the CIFAR-10, CIFAR-100 and HAM10000 datasets, not, however, for all intermediate results.
+The results of the ESC-50 dataset cannot be published at this time, as it would require storing five models per epoch during training in order to store all models of the five-fold cross-validation. This is infeasible on our infrastructure so we only release models for CIFAR-10, CIFAR-100 and HAM10000.
 
 ## Citation
 If you find our work useful or interesting or use our code, please cite our paper as follows