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.7.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | af9f113565c0c09dbefcfb669e2e157c6462901fa498ec04b81ec3f8fb9cb8a4 |
|
MD5 | 955dfcb870e3d9b87e580da23b394fd7 |
|
BLAKE2b-256 | 38a65696aa28e27148bdb34c94e6ba1305c05688f7057cbb86d4dcfb77a16e41 |
Close
Hashes for pytorch_vision_classifier-0.0.7-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 66eadf35fde544c8608e30357e7643e906f2fbca20e69fc5c582b20e785a0e12 |
|
MD5 | 3899ede9dcb9c10e79ea060f70f19c84 |
|
BLAKE2b-256 | 29788a40fc4dc1e07b999be2459e6c77866d8a7559cff0037e1c63727c2e003c |