Stevellen / ResNet-Lightning
PyTorch Lighning wrapper to make training ResNet classifiers easier.
☆29Updated 2 years ago
Alternatives and similar repositories for ResNet-Lightning:
Users that are interested in ResNet-Lightning are comparing it to the libraries listed below
- An education step by step implementation of SimCLR that accompanies the blogpost☆32Updated 3 years ago
- The Objective is to implement Siamese Network for Face Recognition using Pytorch Lightning.☆18Updated 4 years ago
- Examples of using PyTorch hooks, as covered in my YouTube tutorial video.☆33Updated last year
- A Keras implementation of hybrid efficientnet swin transformer model.☆33Updated last year
- ☆24Updated 2 years ago
- Visualizer for PyTorch image models☆44Updated 3 years ago
- LightCollections⚡️: Ready-to-use implementations such as `LightningModules` for various computer vision papers.☆22Updated 2 years ago
- Python Tools for Visual Dataset Transformation☆26Updated 3 weeks ago
- Self-Supervised Learning in PyTorch☆136Updated last year
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆56Updated 2 years ago
- PyTorch implementation of Teacher-Student Network(Knowledge Distillation).☆26Updated 3 years ago
- A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network☆135Updated 4 years ago
- A summarization of Transformer-based architectures for CV tasks, including image classification, object detection, segmentation, and Few-…☆108Updated 2 years ago
- Includes PyTorch -> Keras model porting code for DeiT models with fine-tuning and inference notebooks.☆42Updated 2 years ago
- Easy to use class balanced cross entropy and focal loss implementation for Pytorch☆94Updated 4 months ago
- ☆38Updated 2 years ago
- Simplification of pruned models for accelerated inference | SoftwareX https://doi.org/10.1016/j.softx.2021.100907☆36Updated last month
- A crash course on PyTorch hooks☆40Updated 4 years ago
- A PyTorch Implementation of ViT (Vision Transformer)☆18Updated 3 years ago
- Example unit tests for deep learning projects.☆37Updated 4 years ago
- Experiments with the ideas presented in https://arxiv.org/abs/2003.00152 by Frankle et al.☆29Updated 4 years ago
- PyTorch implementation of "MLP-Mixer: An all-MLP Architecture for Vision" Tolstikhin et al. (2021)☆30Updated 3 years ago
- ModelSoups for Tensorflow2 and Torch☆48Updated 2 years ago
- Comparing four automatic image augmentation techniques in PyTorch: AutoAugment, RandAugment, AugMix, and TrivialAugment☆29Updated 2 years ago
- Deep Learning Experiment Code.☆19Updated 8 months ago
- ☆17Updated 2 years ago
- Vision Transformer GAN Architecture☆12Updated 3 years ago
- Simple MAE (masked autoencoders) with pytorch and pytorch-lightning.☆42Updated last year
- Visualization toolkit for learned features of neural networks in PyTorch. Feature Visualizer, Saliency Map, Guided Gradients, Grad-CAM, D…☆42Updated 3 years ago
- Tensorflow 2.0 Implementation of GCViT: Global Context Vision Transformer☆27Updated last year