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.4.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 8b823243c9ef940fe0653a9238561553b0b645df6b06b76f7948419743445bd0 |
|
MD5 | a6ee3a5c43bd9e161b955f95207222b1 |
|
BLAKE2b-256 | 92471db0f4e11cfd8d1e01d7aee08f9995df22a10aeff3f4b816a50b2c83efd5 |
Close
Hashes for pytorch_vision_classifier-0.0.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e0927396554cf09f9f023c19dd61ad5d203d831d825766877c70fbe7021e82a9 |
|
MD5 | 2e3f1399a0f409cd0a47eb371f8f2624 |
|
BLAKE2b-256 | e2ff343db45d136725ec0889f704e18831b06ebea422d0d24a54bd48ae12281d |