Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
Parameter redundancy from 5 pruning papers through time.
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
# using python file to store data, it's weird I know
table =\
"""year, top-1 before, top-1 after, params before, params after, bibtex_id, title
2015, 31.50, 31.17, 138e6, 10.35e6, han2016deep, Deep Compression
2016, 25.03, 27.83, 21.8e6, 1.93e7, li2016pruning, Pruning Filters for Efficient ConvNets
2016, 31.9, 32.0, 15.19e6, 7.45e6, alvarez2016learning, Learning The Number of Neurons in Deep Networks
2017, 36.69, 36.66, 132.9e6, 23.2e6, liu2017learning, Learning Efficient Convolutional Networks through Network Slimming
2017, 25.02, 26.2, 9.5e6, 4.8e6, huang2017condensenet, CondenseNet: An Efficient DenseNet using Learned Group Convolutions
"""
import csv
from io import StringIO
with StringIO(table) as f:
reader = csv.reader(f, delimiter=',', skipinitialspace=True)
for i, r in enumerate(reader):
if i == 0:
cols = r
pruning = {c:[] for c in cols}
else:
for c,v in zip(cols, r):
pruning[c].append(v if c in ('bibtex_id', 'title') else float(v))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.