overshiki / kmeans_pytorch
pytorch implementation of basic kmeans algorithm(lloyd method with forgy initialization) with gpu support
☆93Updated 6 years ago
Related projects ⓘ
Alternatives and complementary repositories for kmeans_pytorch
- [NeurIPS '18] "Can We Gain More from Orthogonality Regularizations in Training Deep CNNs?" Official Implementation.☆127Updated 2 years ago
- [ICML 2018] "Deep k-Means: Re-Training and Parameter Sharing with Harder Cluster Assignments for Compressing Deep Convolutions"☆150Updated 2 years ago
- Implementation of soft parameter sharing for neural networks☆69Updated 3 years ago
- Code for NeurIPS 2019 Paper, "L_DMI: An Information-theoretic Noise-robust Loss Function"☆117Updated last year
- ☆63Updated 6 years ago
- ☆130Updated last year
- ☆180Updated last year
- Reproduction of Momentum Contrast for Unsupervised Visual Representation Learning☆120Updated 3 months ago
- Implementation of the reversible residual network in pytorch☆101Updated 2 years ago
- Auto-Encoding Transformations (AETv1), CVPR 2019☆108Updated 5 years ago
- Improving Generalization via Scalable Neighborhood Component Analysis☆136Updated last year
- A PyTorch implementation of shake-shake☆110Updated 4 years ago
- PyTorch code for training neural networks without global back-propagation☆162Updated 5 years ago
- Code for Piggyback: Adapting a Single Network to Multiple Tasks by Learning to Mask Weights☆181Updated 5 years ago
- Improving Consistency-Based Semi-Supervised Learning with Weight Averaging☆185Updated 5 years ago
- Implementation and experiments for AdamW on Pytorch☆93Updated 4 years ago
- ☆351Updated 4 years ago
- Tensorflow implementation of S4L: Self-Supervised Semi-Supervised Learning☆96Updated 5 years ago
- An official PyTorch implementation of “Multimodal Model-Agnostic Meta-Learning via Task-Aware Modulation” (NeurIPS 2019) by Risto Vuorio*…☆135Updated 4 years ago
- Utilities for Pytorch☆89Updated 2 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆111Updated 4 years ago
- Pytorch Implementation for CVPR2018 Paper: Learning to Compare: Relation Network for Few-Shot Learning☆251Updated 6 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆98Updated 3 years ago
- Random miniprojects with pytorch.☆173Updated 6 years ago
- Code for 'Low-Shot Learning with Imprinted Weights'☆68Updated 5 years ago
- Code of Empirical Bayes Transductive Meta-Learning with Synthetic Gradients☆58Updated 4 years ago
- PyTorch Implementation of `No Fuss Distance Metric Learning using Proxies`.☆181Updated 4 years ago
- On Network Design Spaces for Visual Recognition☆94Updated 4 years ago
- Why ReLU networks yield high-confidence predictions far away from the training data and how to mitigate the problem [CVPR 2019, oral]☆181Updated 5 years ago
- EnAET: Self-Trained Ensemble AutoEncoding Transformations for Semi-Supervised Learning☆69Updated last year