sony / model_optimization

Model Compression Toolkit (MCT) is an open source project for neural network model optimization under efficient, constrained hardware. This project provides researchers, developers, and engineers advanced quantization and compression tools for deploying state-of-the-art neural networks.
https://sony.github.io/model_optimization/
Apache License 2.0
331 stars 53 forks source link

Pytorch Feature Tutorials Refactor #1225

Closed lior-dikstein closed 1 month ago

lior-dikstein commented 2 months ago

Pull Request Description:

Refactored Pytorch feature notebooks for consistency. Standardized function usage across notebooks. Updated model examples to use pretrained models from torchvision with the ImageNet validation set as the representative dataset (except for the pruning notebook).

Checklist before requesting a review: