added README; update requirements

MingSun-Tse · Dec 17, 2020 · 5381bd1 · 5381bd1
1 parent 58f1773
commit 5381bd1
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -0,0 +1,16 @@
+# Regularization-Pruning
+
+This is the official PyTorch code for our paper [Neural Pruning via Growing Regularization]().
+
+
+# Environment
+
+
+
+# Dataset
+
+
+
+# Run
+
+
diff --git a/requirements.txt b/requirements.txt
@@ -1,9 +1,8 @@
-matplotlib==3.3.2
-torchvision==0.4.2
-numpy==1.19.1
+numpy==1.16.0
 pandas==1.1.2
 torch==1.3.1
-option==1.0.1
-Pillow==7.2.0
-PyYAML==5.3.1
 torchsummary==1.5.1
+matplotlib==3.1.1
+torchvision==0.4.2
+Pillow==8.0.1
+PyYAML==5.3.1