Imported from https://github.com/matthew-mcateer/Keras_pruning