https://github.com/lucadellalib/kmeans-pytorch
A GPU compatible PyTorch implementation of K-means
https://github.com/lucadellalib/kmeans-pytorch
clustering gpu k-means pytorch quantization residual-quantization self-supervised-learning token-extraction
Last synced: 2 months ago
JSON representation
A GPU compatible PyTorch implementation of K-means
- Host: GitHub
- URL: https://github.com/lucadellalib/kmeans-pytorch
- Owner: lucadellalib
- License: apache-2.0
- Created: 2024-08-09T06:40:59.000Z (almost 2 years ago)
- Default Branch: master
- Last Pushed: 2024-09-25T10:44:38.000Z (almost 2 years ago)
- Last Synced: 2025-07-08T13:12:33.842Z (about 1 year ago)
- Topics: clustering, gpu, k-means, pytorch, quantization, residual-quantization, self-supervised-learning, token-extraction
- Language: Python
- Homepage:
- Size: 18.6 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# K-means PyTorch
A GPU-compatible PyTorch implementation of K-means, designed for efficient quantization of representations from large self-supervised models.
---------------------------------------------------------------------------------------------------------
## 🛠️️ Installation
### From source
First of all, install [Python 3.8 or later](https://www.python.org).
Clone or download and extract the repository, navigate to ``, open a
terminal and run:
```
pip install -r requirements
```
---------------------------------------------------------------------------------------------------------
## ▶️ Quickstart
To run the example script, navigate to ``, open a terminal and run
```
python kmeans.py
```
---------------------------------------------------------------------------------------------------------
## 📧 Contact
[luca.dellalib@gmail.com](mailto:luca.dellalib@gmail.com)
---------------------------------------------------------------------------------------------------------