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
Sorting:
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 4 years ago
- A PyTorch implementation of shake-shake☆111Updated 5 years ago
- A plug-in replacement for DataLoader to load Imagenet disk-sequentially in PyTorch.☆239Updated 3 years ago
- Reproduction of Momentum Contrast for Unsupervised Visual Representation Learning☆120Updated 9 months 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
- Full implementation of the paper "Rethinking Softmax with Cross-Entropy: Neural Network Classifier as Mutual Information Estimator".☆101Updated 5 years ago
- ☆182Updated 2 years ago
- Improving Generalization via Scalable Neighborhood Component Analysis☆137Updated last year
- Implementation of the reversible residual network in pytorch☆104Updated 3 years ago
- [ICML 2018] "Deep k-Means: Re-Training and Parameter Sharing with Harder Cluster Assignments for Compressing Deep Convolutions"☆152Updated 3 years ago
- Tensorflow implementation of S4L: Self-Supervised Semi-Supervised Learning☆94Updated 5 years ago
- ☆129Updated 2 years ago
- PyTorch code for training neural networks without global back-propagation☆164Updated 5 years ago
- Implementation and experiments for AdamW on Pytorch☆93Updated 5 years ago
- ☆62Updated 4 years ago
- ☆65Updated 6 years ago
- An official collection of code in different frameworks that reproduces experiments in "Group Normalization"☆118Updated 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
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆112Updated 5 years ago
- Code for our paper "Informative Dropout for Robust Representation Learning: A Shape-bias Perspective" (ICML 2020)☆125Updated 2 years ago
- Distributed, mixed-precision training with PyTorch☆90Updated 4 years ago
- Utilities for Pytorch☆89Updated 2 years ago
- Minimal API for receptive field calculation in PyTorch☆67Updated 2 years ago
- [NeurIPS '18] "Can We Gain More from Orthogonality Regularizations in Training Deep CNNs?" Official Implementation.☆129Updated 3 years ago
- ☆352Updated 5 years ago
- [ICLR'19] Complement Objective Training☆76Updated 6 years ago
- homura is a library for fast prototyping DL research☆107Updated 2 years ago
- Deep Metric Transfer for Label Propagation with Limited Annotated Data☆49Updated last year
- BlockDrop: Dynamic Inference Paths in Residual Networks☆143Updated 2 years ago