fix: add requirements.txt (#1)

HanxunH · Apr 5, 2021 · e6b391e · e6b391e
1 parent d5f1502
commit e6b391e
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 122 deletions.
diff --git a/README.md b/README.md
@@ -1,9 +1,11 @@
 # Unlearnable Examples
-Code for ICLR2021 Paper ["Unlearnable Examples: Making Personal Data Unexploitable "](https://openreview.net/forum?id=iAmZUo0DxC0) by Hanxun Huang, Xingjun Ma, Sarah Monazam Erfani, James Bailey, Yisen Wang.
 
 ## Quick Start
 ##### Use the QuickStart.ipynb notebook for a quick start.
 In the notebook, you can find the minimal implementation for generating sample-wise unlearnable examples on CIFAR-10.
+Please remove `mlconfig` from `models/__init__.py` if you only using the notebook and copy paste the model to the notebook.
+
+
 
 ## Experiments in the paper.
 Check scripts folder for *.sh for each corresponding experiments.
@@ -62,13 +64,3 @@ python3 -u main.py    --version                 resnet18                       \
                       --perturb_tensor_filepath path/to/your/experiment/folder/perturbation.pt \
                       --train
 ```
-
-## Citing this work
-```
-@inproceedings{huang2021unlearnable,
-    title={Unlearnable Examples: Making Personal Data Unexploitable},
-    author={Hanxun Huang and Xingjun Ma and Sarah Monazam Erfani and James Bailey and Yisen Wang},
-    booktitle={ICLR},
-    year={2021}
-}
-```
diff --git a/models/svhn/dense121.yaml b/models/svhn/dense121.yaml
diff --git a/models/svhn/resnet18.yaml b/models/svhn/resnet18.yaml
diff --git a/models/svhn/resnet18_madrys.yaml b/models/svhn/resnet18_madrys.yaml
diff --git a/models/svhn/resnet50.yaml b/models/svhn/resnet50.yaml
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,3 @@
+torch
+torchvision
+mlconfig