efficient_densenet_pytorch
DenseNet optimizer
An implementation of DenseNets optimized to save GPU memory by using checkpointing
A memory-efficient implementation of DenseNets
2k stars
44 watching
327 forks
Language: Python
last commit: over 1 year ago deep-learningdensenetpytorch
Related projects:
| Repository | Description | Stars |
|---|---|---|
 zsef123/efficientnets-pytorch | A PyTorch implementation of EfficientNet for computer vision tasks | 309 |
 bamos/densenet.pytorch | An implementation of a deep neural network architecture in PyTorch | 833 |
 atgambardella/pytorch-es | An implementation of an optimization algorithm for training neural networks in machine learning environments. | 351 |
 rentruewang/koila | A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution | 1,823 |
 4uiiurz1/pytorch-res2net | Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. | 112 |
 leaderj1001/mobilenetv3-pytorch | An implementation of MobileNetV3 using PyTorch with search space optimization | 292 |
 robotlocomotion/pytorch-dense-correspondence | This project presents a method for learning dense object descriptors from self-supervised data, enabling tasks such as robotic manipulation and object recognition. | 561 |
 anjiecheng/mnasnet-pytorch | An implementation of a neural network architecture search algorithm optimized for mobile devices. | 271 |
 xternalz/wideresnet-pytorch | An implementation of Wide Residual Networks in PyTorch for efficient deep learning on CIFAR10/100 datasets. | 334 |
 eduardoleao052/js-pytorch | A JavaScript library that provides GPU-accelerated deep learning capabilities with automatic differentiation and neural network layers. | 1,093 |
 rdspring1/pytorch_gbw_lm | Trains a large-scale PyTorch language model on the 1-Billion Word dataset | 123 |
 locuslab/optnet | A PyTorch module that adds differentiable optimization as a layer to neural networks | 517 |
 titu1994/densenet | An implementation of a deep learning network architecture with dense connectivity and bottleneck layers. | 707 |
 zhanghang1989/pytorch-encoding | A Python framework for building deep learning models with optimized encoding layers and batch normalization. | 2,044 |
 neuralmagic/sparseml | Enables the creation of smaller neural network models through efficient pruning and quantization techniques | 2,083 |