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
- Code for NeurIPS 2019 Paper, "L_DMI: An Information-theoretic Noise-robust Loss Function"☆119Updated last year
- Implementation of soft parameter sharing for neural networks☆69Updated 4 years ago
- [NeurIPS '18] "Can We Gain More from Orthogonality Regularizations in Training Deep CNNs?" Official Implementation.☆128Updated 3 years ago
- ☆65Updated 6 years ago
- Implementation and experiments for AdamW on Pytorch☆93Updated 5 years ago
- Improving Generalization via Scalable Neighborhood Component Analysis☆137Updated last year
- Tensorflow implementation of S4L: Self-Supervised Semi-Supervised Learning☆95Updated 5 years ago
- [ICLR 2019] ProbGAN: Towards Probabilistic GAN with Theoretical Guarantees☆32Updated 5 years ago
- Full implementation of the paper "Rethinking Softmax with Cross-Entropy: Neural Network Classifier as Mutual Information Estimator".☆101Updated 5 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 4 years ago
- Deep Metric Transfer for Label Propagation with Limited Annotated Data☆49Updated last year
- A PyTorch implementation of shake-shake☆111Updated 4 years ago
- ☆130Updated 2 years ago
- PyTorch Implementations of Dropout Variants☆87Updated 7 years ago
- EnAET: Self-Trained Ensemble AutoEncoding Transformations for Semi-Supervised Learning☆81Updated last year
- Reproduction of Momentum Contrast for Unsupervised Visual Representation Learning☆120Updated 8 months ago
- [NeurIPS'20] GradAug: A New Regularization Method for Deep Neural Networks☆93Updated 4 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
- Distributed, mixed-precision training with PyTorch☆89Updated 4 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆98Updated 4 years ago
- PyTorch implementation of shake-shake regularization☆48Updated 4 years ago
- Improving Consistency-Based Semi-Supervised Learning with Weight Averaging☆186Updated 6 years ago
- Evaluating AlexNet features at various depths☆39Updated 4 years ago
- Pytorch implementation of "A Simple Framework for Contrastive Learning of Visual Representations"☆81Updated 11 months ago
- Learning Metrics from Teachers: Compact Networks for Image Embedding (CVPR19)☆76Updated 5 years ago
- Minimal API for receptive field calculation in PyTorch☆67Updated 2 years ago
- Implementation of the reversible residual network in pytorch☆104Updated 3 years ago
- A pytorch dataset sampler for always sampling balanced batches.☆113Updated 4 years ago
- ICML 2019: Understanding and Utilizing Deep Neural Networks Trained with Noisy Labels☆91Updated 4 years ago