From 8b97a83596cb39bc7f52a409bba324273b68e73c Mon Sep 17 00:00:00 2001
From: Yaxin Li <49900442+I-am-Bot@users.noreply.github.com>
Date: Fri, 8 May 2020 00:52:49 -0400
Subject: [PATCH] Update README.md

---
 README.md | 1 +
 1 file changed, 1 insertion(+)

diff --git a/README.md b/README.md
index 7abeac7f..3856dc81 100644
--- a/README.md
+++ b/README.md
@@ -62,6 +62,7 @@ python examples/graph/test_mettack.py --dataset cora --ptb_rate 0.05
     ```python
     from deeprobust.image.attack.pgd import PGD
     from deeprobust.image.config import attack_params
+    import torch
     import deeprobust.image.netmodels.resnet as resnet
     
     model = resnet.ResNet18().to('cuda')