Wraper for pytorch lighining classification
Project description
LitClassifier
How to use
git clone https://github.com/anhvth/litclassifier
pip install -e ./litclassifier
Trainer example
lit_trainer --help
# Demo training with simple mnist model
# To change model please take a look at `configs/00_mnist_vanila.py`
lit_trainer configs/00_mnist_vanila.py
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
lit_classifier-0.0.2.tar.gz
(23.0 kB
view hashes)
Built Distribution
Close
Hashes for lit_classifier-0.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | a5636b305b5757da5e340c17a5fdff7adcdd6ad5ef115ab659965abd26c90997 |
|
MD5 | 2bc6ddd5b02094d6418907479ae9ee80 |
|
BLAKE2b-256 | 052a1579cbc52c58bc3fdf67cd71655680ef3b65b2599a15b80d21892359335f |