kuangliu / pytorch-training-templateLinks
PyTorch model training template
☆24Updated 5 years ago
Alternatives and similar repositories for pytorch-training-template
Users that are interested in pytorch-training-template are comparing it to the libraries listed below
Sorting:
- Accuracy, precision, recall, confusion matrix computation with batch updates☆14Updated 5 years ago
- My solution to the Global Data Science Challenge☆36Updated 5 years ago
- State-of-the-art data augmentation search algorithms in PyTorch☆47Updated 2 years ago
- Experiments with the ideas presented in https://arxiv.org/abs/2003.00152 by Frankle et al.☆29Updated 5 years ago
- Light Face Detection using PyTorch Lightning☆83Updated 2 years ago
- Swish Activation - PyTorch CUDA Implementation☆37Updated 6 years ago
- For ICDAR 2019 Paper on End-to-end License Plate and Scene Text Recognition with multi-head attention models☆25Updated 4 years ago
- TF 2 implementation Learning to Resize Images for Computer Vision Tasks (https://arxiv.org/abs/2103.09950v1).☆53Updated 4 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆86Updated 4 years ago
- Knowledge Distillation Algorithms implemented with PyTorch☆17Updated 6 years ago
- This is the code base for paper "G-SimCLR : Self-Supervised Contrastive Learning with Guided Projection via Pseudo Labelling" by Souradip…☆79Updated 4 years ago
- (WACV 2020) Tensorflow implementation of Boosting Standard Classification Architectures Through a Ranking Regularizer☆23Updated 4 years ago
- ☆23Updated 6 years ago
- TF2 implementation of knowledge distillation using the "function matching" hypothesis from https://arxiv.org/abs/2106.05237.☆88Updated 4 years ago
- Pytorch implementation of Quadratic Additive Angular Margin Loss for Face Recognition☆35Updated 5 years ago
- A visualization and transformation of pytorch model☆30Updated 6 years ago
- Few training heuristics and small architectural changes that can significantly improve YOLOv3 performance with tiny increase in inference…☆12Updated 5 years ago
- Implementation of Online Label Smoothing in PyTorch☆96Updated 3 years ago
- Implementation of Spectral Leakage and Rethinking the Kernel Size in CNNs in Pytorch☆14Updated 4 years ago
- Implementation of various Vision Transformers I found interesting☆84Updated 4 years ago
- ☆17Updated 5 years ago
- Keras Implementation of MixNets: Mixed Depthwise Convolutions☆40Updated 5 years ago
- This research is carried out to find the necessity of fully connected (FC) layers in CNN for image classification.☆14Updated 6 years ago
- re-implement for paper: SqueezeNext: Hardware-Aware Neural Network Design. (SqueezeNext)☆29Updated 7 years ago
- Vision Transformer using TensorFlow 2.0☆99Updated 5 years ago
- Pytorch implementation of Robust Bi-Tempered Logistic Loss Based on Bregman Divergences☆62Updated 6 years ago
- PyTorch implementation of the paper The Lottery Ticket Hypothesis for Object Recognition☆23Updated 4 years ago
- Our implementation of Shampoo optimizer based on https://arxiv.org/pdf/1802.09568.pdf☆12Updated 6 years ago
- batchboost is a variation on MixUp that instead of mixing just two images, mixes many images together.☆44Updated 6 years ago
- Learning computer vision by striving to maximise accuracy on the Stanford Cars dataset☆57Updated 6 years ago