netspresso-trainer 0.1.2

NetsPresso Python Package

Start training models (including ViTs) with NetsPresso Trainer, compress and deploy your model with NetsPresso!

Website • Getting Started • Issues • Docs

---

Table of contents

• Getting started
• Installation

Getting started

Write your training script in train.py like:

from netspresso_trainer import train_cli

if __name__ == '__main__':
    logging_dir = train_cli()
    print(f"Training results are saved at: {logging_dir}")

Then, train your model with your own configuraiton:

python train.py\
  --data config/data/beans.yaml\
  --augmentation config/augmentation/classification.yaml\
  --model config/model/resnet/resnet50-classification.yaml\
  --training config/training/classification.yaml\
  --logging config/logging.yaml\
  --environment config/environment.yaml

Or you can start NetsPresso Trainer by just executing console script which has same feature.

netspresso-train\
  --data config/data/beans.yaml\
  --augmentation config/augmentation/classification.yaml\
  --model config/model/resnet/resnet50-classification.yaml\
  --training config/training/classification.yaml\
  --logging config/logging.yaml\
  --environment config/environment.yaml

Please refer to scripts/example_train.sh.

NetsPresso Trainer is compatible with NetsPresso service. We provide NetsPresso Trainer tutorial that contains whole procedure from model train to model compression and benchmark. Please refer to our colab tutorial.

Installation

Prerequisites

• Python 3.8 | 3.9 | 3.10
• PyTorch 1.13.0 (recommended) (compatible with: 1.11.x - 1.13.x)

Install with pypi (stable)

pip install netspresso_trainer

Install with GitHub

pip install git+https://github.com:Nota-NetsPresso/netspresso-trainer.git@stable

To install with editable mode,

git clone https://github.com:Nota-NetsPresso/netspresso-trainer.git .
pip install -e netspresso-trainer

Set-up with docker

Please clone this repository and refer to Dockerfile and docker-compose-example.yml.
For docker users, we provide more detailed guide in our Docs.

Tensorboard

We provide basic tensorboard to track your training status. Run the tensorboard with the following command:

tensorboard --logdir ./outputs --port 50001 --bind_all

where PORT for tensorboard is 50001.
Note that the default directory of saving result will be ./outputs directory.

Pretrained weights

Please refer to our official documentation for pretrained weights supported by NetsPresso Trainer.