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
- A PyTorch implementation of shake-shake☆111Updated 4 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 and experiments for AdamW on Pytorch☆93Updated 4 years ago
- A plug-in replacement for DataLoader to load Imagenet disk-sequentially in PyTorch.☆238Updated 3 years ago
- ☆63Updated 6 years ago
- Reproduction of Momentum Contrast for Unsupervised Visual Representation Learning☆120Updated 3 months ago
- [NeurIPS '18] "Can We Gain More from Orthogonality Regularizations in Training Deep CNNs?" Official Implementation.☆127Updated 2 years ago
- PyTorch Implementations of Dropout Variants☆87Updated 6 years ago
- (ICCV 2019 Oral) Many Task Learning With Task Routing http://openaccess.thecvf.com/content_ICCV_2019/html/Strezoski_Many_Task_Learning_Wi…☆62Updated 4 years ago
- Utilities for Pytorch☆89Updated 2 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"☆117Updated last year
- Metric learning models in PyTorch with results on CUB2011, CARS196, Stanford Online Products☆65Updated 2 years ago
- Implementation of the reversible residual network in pytorch☆101Updated 2 years ago
- Implementation of soft parameter sharing for neural networks☆69Updated 3 years ago
- PyTorch code for softmax variants: center loss, cosface loss, large-margin gaussian mixture, COCOLoss, ring loss☆252Updated 6 years ago
- A Pytorch implementation of "LegoNet: Efficient Convolutional Neural Networks with Lego Filters" (ICML 2019).☆140Updated 4 years ago
- Code of Empirical Bayes Transductive Meta-Learning with Synthetic Gradients☆58Updated 4 years ago
- An official collection of code in different frameworks that reproduces experiments in "Group Normalization"☆117Updated 3 years ago
- Pytorch implementation of group normalization in https://arxiv.org/abs/1803.08494 (Following the PyTorch Style)☆87Updated 5 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 3 years ago
- Improving Generalization via Scalable Neighborhood Component Analysis☆136Updated last year
- [ICLR'19] Complement Objective Training☆76Updated 5 years ago
- Improving Consistency-Based Semi-Supervised Learning with Weight Averaging☆185Updated 5 years ago
- ☆61Updated 4 years ago
- Full implementation of the paper "Rethinking Softmax with Cross-Entropy: Neural Network Classifier as Mutual Information Estimator".☆100Updated 4 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆112Updated 4 years ago
- Auto-Encoding Transformations (AETv1), CVPR 2019☆108Updated 5 years ago
- homura is a library for fast prototyping DL research☆106Updated 2 years ago
- EnAET: Self-Trained Ensemble AutoEncoding Transformations for Semi-Supervised Learning☆81Updated last year