densechen / kmeans-gpuLinks
kmeans-gpu with pytorch support
☆13Updated 4 years ago
Alternatives and similar repositories for kmeans-gpu
Users that are interested in kmeans-gpu are comparing it to the libraries listed below
Sorting:
- [ICML2023] InfoOT: Information Maximizing Optimal Transport☆41Updated 2 years ago
- Unsupervised Part Discovery from Contrastive Reconstruction (NeurIPS 2021)☆66Updated 2 years ago
- Robustness via Cross-Domain Ensembles, ICCV 2021 [Oral]☆40Updated 4 years ago
- Code for the ECCV22 paper Demystifying Unsupervised Semantic Correspondence Estimation☆14Updated 3 years ago
- Robust Optimal Transport code☆44Updated 3 years ago
- The repository for Hyperbolic Representation Learning for Computer Vision, ECCV 2022☆66Updated 3 years ago
- CVPR 2020, Semantic Correspondence as an Optimal Transport Problem, Pytorch Implementation.☆104Updated 4 years ago
- [CVPR'21] SetVAE: Learning Hierarchical Composition for Generative Modeling of Set-Structured Data, in PyTorch☆78Updated 3 years ago
- B-cos Networks: Alignment is All we Need for Interpretability☆114Updated 2 years ago
- The official github repo for "Test-Time Training with Masked Autoencoders"☆92Updated 2 years ago
- Official implementation of "Go with the Flows: Mixtures of Normalizing Flows for Point Cloud Generationand Reconstruction"☆12Updated 4 years ago
- PyTorch implementation of the paper "No reason for no supervision: Improving the generalization of supervised models"☆18Updated 2 years ago
- Official code for "Self-supervised learning with rotation-invariant kernels"☆12Updated 2 years ago
- Learning low-shot object classification with explicit shape bias learned from point clouds☆54Updated 4 years ago
- ☆48Updated 2 years ago
- Autoencoding Variational Transformations (AVT) in pytorch, ICCV 2019☆18Updated 6 years ago
- MoCo with Alignment and Uniformity Loss.☆63Updated 4 years ago
- ☆19Updated last year
- Backpropagation-Friendly-Eigendecomposition☆73Updated 6 years ago
- Code Release for "Self-supervised Learning is More Robust to Dataset Imbalance"☆39Updated 3 years ago
- [NeurIPS] TTT++: When Does Self-supervised Test-time Training Fail or Thrive?☆73Updated 3 years ago
- NeurIPS22 "RankFeat: Rank-1 Feature Removal for Out-of-distribution Detection" and T-PAMI Extension☆20Updated 11 months ago
- ☆18Updated 4 years ago
- PyTorch implementation of Metric-Guided Prototype Learning for hierarchical classification.☆28Updated 4 years ago
- Code and instructions accompanying ICCV'23 paper Protoype-based Dataset Comparison☆18Updated 2 years ago
- Implementation of our ICLR2023 paper "Spherical-Sliced Wasserstein"☆14Updated last year
- NeurIPS23 "Flow Factorized Representation Learning"☆41Updated last month
- Implementation of the Gumbel-Sigmoid distribution in PyTorch.☆21Updated 3 years ago
- Visual Representation Learning Benchmark for Self-Supervised Models☆35Updated last year
- ☆42Updated 4 years ago