Skip to content

Olive-ai 0.3.1

Compare
Choose a tag to compare
@leqiao-1 leqiao-1 released this 21 Aug 07:55
· 1 commit to rel-0.3.1 since this release

Examples

The following examples are added

Passes (optimization techniques)

  • Introduce TorchTRTConversion
  • Introduce SparseGPT pass for one-shot model pruning on large GPT like models using the algorithm proposed in https://arxiv.org/abs/2301.00774.

Systems

  • Add AzureML sku support for AMLSystem

Evaluator

  • Add metric_func config to custom metric. Olive will run the inference for custom eval func for user. User doesn't need to do inference by themselves.
  • Add RawDataContainer:
    SNPE evaluation and quantization now accept generic dataloaders such as torch dataloader

Metrics

  • Add Perplexity metric for text-generation task

Engine

  • Provide the interface to let user set the multi pass flows to run in save olive workflow