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.6.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 62961dcc67118d3561486717cdb382fbf2a1177819f7949d31ba8f2ce2f7ee07 |
|
MD5 | 680e71f6cac328c8aad1a04b3d427977 |
|
BLAKE2b-256 | c3e2b260dff00b9e5adbbe2508fd6228b586d727985532df99c123563ed49e6e |
Close
Hashes for pytorch_vision_classifier-0.0.6-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c0bd4641a72c64da9cd02ffe3b4e5de99945bee73c37b0e8a42197bf4bba4807 |
|
MD5 | 3b9a6db5eabbe1e3a571b7698a957db1 |
|
BLAKE2b-256 | 4afb1635ac58f93e8e2880b393b809e34507a3a00358729552f3bc5789d5e472 |