Update README.md

tailintalent · May 8, 2020 · 8b97a83 · 8b97a83
1 parent e963cfb
commit 8b97a83
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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')