← all repositories

BenWhetton/keras-surgeon

A Python library for surgical modification of trained Keras models, primarily used for deep neural network pruning.

408 stars Python ML Frameworks
keras-surgeon
Velocity · 7d
+0.1
★ / day
Trend
steady
star history

Keras-surgeon provides utilities for modifying trained Keras models without retraining. It enables operations like deleting neurons or channels from layers, removing or inserting entire layers, and replacing layers. The library works with any model architecture and supports batch modifications via a Surgeon class. It is compatible with both tensorflow.keras and standalone keras implementations.

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