overshiki / kmeans_pytorch
pytorch implementation of basic kmeans algorithm(lloyd method with forgy initialization) with gpu support
☆93Updated 6 years ago
Alternatives and similar repositories for kmeans_pytorch:
Users that are interested in kmeans_pytorch are comparing it to the libraries listed below
- [ICML 2018] "Deep k-Means: Re-Training and Parameter Sharing with Harder Cluster Assignments for Compressing Deep Convolutions"☆152Updated 3 years ago
- [NeurIPS '18] "Can We Gain More from Orthogonality Regularizations in Training Deep CNNs?" Official Implementation.☆129Updated 3 years ago
- A PyTorch implementation of shake-shake☆111Updated 5 years ago
- Implementation of the reversible residual network in pytorch☆104Updated 3 years ago
- Implementation and experiments for AdamW on Pytorch☆93Updated 5 years ago
- ☆65Updated 6 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 4 years ago
- A plug-in replacement for DataLoader to load Imagenet disk-sequentially in PyTorch.☆239Updated 3 years ago
- Implementation of soft parameter sharing for neural networks☆69Updated 4 years ago
- Chainer implementation of deep-INFOMAX☆34Updated 6 years ago
- Tensorflow implementation of S4L: Self-Supervised Semi-Supervised Learning☆94Updated 5 years ago
- Full implementation of the paper "Rethinking Softmax with Cross-Entropy: Neural Network Classifier as Mutual Information Estimator".☆101Updated 5 years ago
- Why ReLU networks yield high-confidence predictions far away from the training data and how to mitigate the problem [CVPR 2019, oral]☆182Updated 5 years ago
- Code for NeurIPS 2019 Paper, "L_DMI: An Information-theoretic Noise-robust Loss Function"☆119Updated last year
- homura is a library for fast prototyping DL research☆107Updated 2 years ago
- ☆182Updated 2 years ago
- Auto-Encoding Transformations (AETv1), CVPR 2019☆108Updated 5 years ago
- An official collection of code in different frameworks that reproduces experiments in "Group Normalization"☆118Updated 4 years ago
- [ICLR 2019] ProbGAN: Towards Probabilistic GAN with Theoretical Guarantees☆32Updated 5 years ago
- PyTorch Implementations of Dropout Variants☆87Updated 7 years ago
- PyTorch implementation of shake-drop regularization☆54Updated 5 years ago
- An implementation of shampoo☆74Updated 7 years ago
- ☆352Updated 5 years ago
- Code of Empirical Bayes Transductive Meta-Learning with Synthetic Gradients☆58Updated 5 years ago
- (ICCV 2019 Oral) Many Task Learning With Task Routing http://openaccess.thecvf.com/content_ICCV_2019/html/Strezoski_Many_Task_Learning_Wi…☆64Updated 5 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆112Updated 5 years ago
- ☆62Updated 4 years ago
- Tensorflow implementation for "Local Aggregation for Unsupervised Learning of Visual Embeddings"☆59Updated 4 years ago
- Deep Learning under Privileged Information Using Heteroscedastic Dropout (CVPR 2018, Official Repo)☆38Updated 4 years ago
- Reproduction of Momentum Contrast for Unsupervised Visual Representation Learning☆120Updated 9 months ago