This library is to help you train and evaluate PyTorch classification model easily and quickly
Project description
pytorch-vision-classifier
The main target of this template is to help during building your classification model where:
- It helps load the dataset whether the dataset in one directory or multiple directories
- It shows the dataset details once the dataset directory has been specified
- It helps know the available GPU devices and pick one to use easily
- It helps get a pre-trained model for you with the last layer updated with or without a dropout layer and initialized by an algorithm you choose from the most common initialization algorithms
- It helps know the GPU memory usage of your model
- It helps understand the timing of different steps during model training
- It helps find the best learning rate using the algorithm published by Leslie N. Smith in the paper Cyclical Learning Rates for Training Neural Networks. The original code
- It provides a dashboard to monitor your model during the training process
- It helps track the metric that you choose to find the best model for your problem
- It provides a compressed version of your model to be used for deployment purpose
In order to install, you need to download pytorch. then open the command prompt and type:
pip install pytorch_vision_classifier
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for pytorch_vision_classifier-0.0.1.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4cb4120bc2b7bebd3476b6d3260af92ddef5665979dff55840dc36ec6d55189e |
|
MD5 | f0659f7b9247e9ae69dd9385dd76967c |
|
BLAKE2b-256 | c67bebc2410479e01de15b55dcbc59c3179cff2a135eb55b5edf52c4076457ce |
Close
Hashes for pytorch_vision_classifier-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c77a297f8f67fb77db0b412ad07e932f8a0843aa93b66ff9acc8145c64dfb0da |
|
MD5 | d18d0e8694d2ace8899ca64eca531e24 |
|
BLAKE2b-256 | ed6beaa5ab904e953e8f3e8345e2199dda7288d5f69515329e06bd8c2ce0f94c |