Tensorflow-pruning-benchmark

Using the Magnitude-based weight pruning with Keras based API, benchmark availables datasets from tf.keras.datasets

Get started

Prerequisites

    # Using miniconda
    conda install -c conda-forge notebook
    pip install -q -U tensorflow-gpu==1.14.0
    pip install -q tensorflow-model-optimization

Notebooks

    cd ./notebooks
    jupyter notebook

test