Examples

More examples can be found in our GitHub repository.

HPO Quickstart with PyTorch

Use HPO to tune a PyTorch FashionMNIST model

HPO

HPO Quickstart with PyTorch

HPO Quickstart with TensorFlow

Use HPO to tune a TensorFlow MNIST model

HPO

HPO Quickstart with TensorFlow

HPO using command line tool

Run HPO experiment with nnictl

HPO

Run HPO Experiment with nnictl

Hello, NAS!

Beginners' NAS tutorial on how to search for neural architectures for MNIST dataset.

NAS

Hello, NAS!

Use NAS Benchmarks as Datasets

Query data from popular NAS benchmarks from our preprocessed benchmark database.

NAS

Use NAS Benchmarks as Datasets

Get Started with Model Pruning on MNIST

Familiarize yourself with pruning to compress your model

Compression

Pruning Quickstart

Get Started with Model Quantization on MNIST

Familiarize yourself with quantization to compress your model

Compression

Quantization Quickstart

Speedup Model with Mask

Make your model real smaller and faster with speed-up after pruned by pruner

Compression

Speedup Model with Mask

Speedup Model with Calibration Config

Make your model real smaller and faster with speed-up after quantized by quantizer

Compression

Speed Up Quantized Model with TensorRT

Pruning Bert on Task MNLI

An end to end example for how to using NNI pruning transformer and show the real speedup number

Compression

Pruning Bert on Task MNLI

Quantize Bert on Task MNLI

An end to end example for how to using NNI to simulate the process of quantize transformer

Compression

Quantize BERT on Task GLUE