BenWhetton/keras-surgeon
A Python library for surgical modification of trained Keras models, primarily used for deep neural network pruning.

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.