diff --git a/README.md b/README.md index f25c85d..cd3e57e 100644 --- a/README.md +++ b/README.md @@ -82,11 +82,9 @@ This repository has been created with the idea of providing a simple skeleton fr start a PyTorch Lightning project. Instead of favoring the customizability, we favored the simplicity and we intended this template as a base for building more specific templates based on the user needs (for example by forking this one). However, there are several other repositories with different -features that you can check if interested. We will list two of them here: +features that you can check if interested. We will list one of them here: - [lucmos/nn-template](https://github.com/lucmos/nn-template): a very nice template with support for DVC. -- [hobogalaxy/lightning-hydra-template](https://github.com/hobogalaxy/lightning-hydra-template): - another useful and very well documented template repository. ## How to use it @@ -164,4 +162,4 @@ To fine-tune any of the 4 base sentence encoders you should follow these steps. 1. `supervised-intent-clustering/conf/model/default_model.yaml` 2. `supervised-intent-clustering/conf/train/default_train.yaml` 3. `supervised-intent-clustering/conf/data/default_data.yaml` - 4. `supervised-intent-clustering/src/train.py` \ No newline at end of file + 4. `supervised-intent-clustering/src/train.py` diff --git a/requirements.txt b/requirements.txt index 2fc1d1e..0190e15 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch==2.2.0 +torch==2.8.0 sentence-transformers hydra-core tqdm