← all repositories

open-mmlab/mmrazor

A PyTorch-based toolbox for compressing deep learning models through pruning, quantization, knowledge distillation, and neural architecture search.

1.7k stars Python ML Frameworks
mmrazor
Velocity · 7d
+1.0
★ / day
Trend
steady
star history

This repository is part of the OpenMMLab ecosystem and provides a comprehensive model compression toolkit. It supports multiple compression techniques including network pruning, quantization, knowledge distillation, and neural architecture search (NAS). Built on PyTorch, it offers tools for compressing models across computer vision tasks like classification, detection, and segmentation, with support for methods such as AutoSlim and DARTS.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.