jokofa / torch_kmeansLinks
PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.
☆65Updated 2 years ago
Alternatives and similar repositories for torch_kmeans
Users that are interested in torch_kmeans are comparing it to the libraries listed below
Sorting:
- The code of "Improving Deep Regression with Ordinal Entropy" in ICLR 2023☆51Updated last year
- The official github repo for "Test-Time Training with Masked Autoencoders"☆83Updated last year
- This is the official GitHub for paper: On the Versatile Uses of Partial Distance Correlation in Deep Learning, in ECCV 2022☆173Updated 2 years ago
- [ICLR 2022] "Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practice" by Peihao Wang, Wen…☆79Updated last year
- This is the official implementation for Image Clustering via the Principle of Rate Reduction in the Age of Pretrained Models.☆28Updated last year
- Visualizing representations with diffusion based conditional generative model.☆95Updated 2 years ago
- Transformers w/o Attention, based fully on MLPs☆93Updated last year
- ☆45Updated 2 years ago
- [NeurIPS'23] DropPos: Pre-Training Vision Transformers by Reconstructing Dropped Positions☆60Updated last year
- [CVPR 2024] Friendly Sharpness-Aware Minimization☆33Updated 7 months ago
- Learnable Fourier Features for Multi-Dimensional Spatial Positional Encoding☆49Updated 8 months ago
- ☆22Updated 4 years ago
- A torch-based implementation of K-Means and K-Means++☆17Updated 4 years ago
- A collection of differentiable SVD methods and ICCV21 "Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance P…☆77Updated last year
- A pure PyTorch implementation of kmeans and GMM with distributed clustering.☆46Updated 2 years ago
- Official PyTorch code for "Out-of-distribution detection with denoising diffusion models"☆50Updated 11 months ago
- Official implementation of AAAI 2023 paper "Parameter-efficient Model Adaptation for Vision Transformers"☆105Updated last year
- List of papers that combine self-supervision and continual learning☆65Updated 2 months ago
- ☆152Updated 2 years ago
- Hyperbolic Image Segmentation, CVPR 2022☆86Updated 2 years ago
- Official Implementation of the CrossMAE paper: Rethinking Patch Dependence for Masked Autoencoders☆113Updated last month
- (ICLR 2023) Official PyTorch implementation of "What Do Self-Supervised Vision Transformers Learn?"☆111Updated last year
- Using pre-trained Diffusion models as priors for inference tasks☆203Updated 2 years ago
- Differentiable Top-k Classification Learning☆82Updated 2 years ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆145Updated 3 years ago
- Official implementation of "A simple, efficient and scalable contrastive masked autoencoder for learning visual representations".☆32Updated 2 years ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated 2 years ago
- Official Code for NeurIPS 2022 Paper: How Mask Matters: Towards Theoretical Understandings of Masked Autoencoders☆58Updated last year
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- PyTorch reimplementation of FlexiViT: One Model for All Patch Sizes☆62Updated last year