pytorch-cnn-finetune
CNN fine-tuner
A PyTorch-based framework for fine-tuning pre-trained convolutional neural networks on various architectures and datasets.
Fine-tune pretrained Convolutional Neural Networks with PyTorch
724 stars
18 watching
123 forks
Language: Python
last commit: 3 months ago convolutional-neural-networksdeep-learningpretrained-modelspytorch
Related projects:
| Repository | Description | Stars |
|---|---|---|
 spandan-madan/pytorch_fine_tuning_tutorial | Provides guidance on fine-tuning pre-trained models for image classification tasks using PyTorch. | 279 |
 jacobgil/pytorch-pruning | This project provides a PyTorch implementation of pruning techniques to reduce the computational resources required for neural network inference. | 875 |
 randl/shufflenetv2-pytorch | An implementation of a lightweight convolutional neural network architecture for mobile devices | 191 |
 graal-research/poutyne | A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities | 569 |
 longcw/faster_rcnn_pytorch | An implementation of Faster R-CNN using PyTorch | 1,724 |
 xternalz/wideresnet-pytorch | An implementation of Wide Residual Networks in PyTorch for efficient deep learning on CIFAR10/100 datasets. | 333 |
 guillaume-chevalier/hyperopt-keras-cnn-cifar-100 | Automates hyperparameter optimization and neural network architecture search using Hyperopt on a CNN model for the CIFAR-100 dataset | 106 |
 erotemic/netharn | A PyTorch framework for managing and automating deep learning training loops with features like hyperparameter tracking and single-file deployments. | 39 |
 utkuozbulak/pytorch-cnn-adversarial-attacks | PyTorch implementation of various Convolutional Neural Network adversarial attack techniques | 350 |
 kaiyangzhou/dassl.pytorch | A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. | 1,217 |
 zsef123/efficientnets-pytorch | A PyTorch implementation of EfficientNet for computer vision tasks | 309 |
 atgambardella/pytorch-es | An implementation of an optimization algorithm for training neural networks in machine learning environments. | 350 |
 norse/norse | Deep learning framework for spiking neural networks | 673 |
 4uiiurz1/pytorch-res2net | Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. | 112 |
 shawn1993/cnn-text-classification-pytorch | An implementation of Kim's Convolutional Neural Networks for Sentence Classification in PyTorch | 1,020 |