Skip to content

antonioverdi/MLReproChallenge

Repository files navigation

ML Reproducibility Challenge 2020

Description

This repository contains all of the code used to replicate the results of the ICLR 2020 conference paper

"Comparing Rewinding and Fine-Tuning in Neural Network Pruning"

It also includes code for unstructured sensitivity based pruning from: "SNIP: Single-Shot Network Pruning Based on Connection Sensitivity"

Requirements/Setup

Instructions for Running Code

Architectures and Datasets

Dataset Network Optimizer Learning Rate
CIFAR 10 ResNet-56 Nesterov SGD

Batch Size: 128
Weight Decay: 0.0002
Epochs: 182


ImageNet ResNet-34 Nesterov SGD

Batch Size: 1024
Weight Decay: 0.0001
Epochs: 90
ImageNet Resnet-50 Nesterov SGD

Batch Size: 1024
Weight Decay: 0.0001
Epochs: 90
WMT16 EN-DE GNMT Adam


Batch Size: 2048
Epochs: 5

About

Machine Learning Reproducibility Challenge 2020

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published