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.3.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 721c9ba6c439d28456064b583d9dddf08d1ee699637dca9c11a61df44329ad3a |
|
MD5 | 10ea96a9900a0b04eaea09238fc51ae4 |
|
BLAKE2b-256 | 0f892a469e7d5747fc7757b4b5734189070f4fb3e2e4ac7c90ebf108769f22fa |
Close
Hashes for pytorch_vision_classifier-0.0.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 36edb97b8955be0dd112f5c1862d31c276696297f0fc17578f87fb1f6fc98a76 |
|
MD5 | f1c9e1a55a198c67a0600d98a6077adb |
|
BLAKE2b-256 | 5c3b83c27cbdd01e1a56c06e3431d041e5d297137b3cb2eed59cb15d8b5cb540 |