https://github.com/tchaton/keras_pruner

This repository contains part of the code to implement an optimized pruning algorithm in keras
https://github.com/tchaton/keras_pruner

Last synced: over 1 year ago
JSON representation

This repository contains part of the code to implement an optimized pruning algorithm in keras

• Host: GitHub
• URL: https://github.com/tchaton/keras_pruner
• Owner: tchaton
• Created: 2018-05-01T14:15:05.000Z (about 8 years ago)
• Default Branch: master
• Last Pushed: 2018-05-01T14:31:05.000Z (about 8 years ago)
• Last Synced: 2025-01-21T03:24:52.551Z (over 1 year ago)
• Language: Python
• Size: 30.3 KB
• Stars: 3
• Watchers: 3
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# Keras_Pruner

## Efficient Keras implementation of  [\[1611.06440 Pruning Convolutional Neural Networks for Resource Efficient Inference\]](https://arxiv.org/abs/1611.06440) ##

# Only the code of the pruner is present.

# Process :

  - Create pipeline using Augmentor on my repo

  - Feed the trained neural network to the Transformer (wrap layers into a custom Wrapper)

  - Feed this model to NVIDIA_Pruner, set shapes (#could be implemented inside this class)

  - WAIT FOR PRUNING