fix model import in dat

README.md

@@ -31,7 +31,7 @@
 
 ## Our Research Project
 
-- **[CVPR 2023]** ImageNet-E: Benchmarking Neural Network Robustness via Attribute Editing [[Paper](https://arxiv.org/abs/2210.04213), [Image editing toolkit](ImageNet-Editing), [ImageNet-E dataset](https://drive.google.com/file/d/19M1FQB8c_Mir6ermRsukTQReI-IFXeT0/view?usp=sharing)]
+- **[CVPR 2023]** ImageNet-E: Benchmarking Neural Network Robustness via Attribute Editing [[Paper](https://openreview.net/forum?id=C1A2HD6EEGO), [Image editing toolkit](benchmarks/imagenet-e/ImageNet-Editing), [ImageNet-E dataset](https://drive.google.com/file/d/19M1FQB8c_Mir6ermRsukTQReI-IFXeT0/view?usp=sharing)]
 - **[ICLR 2023]** Inequality phenomenon in $l_\infty$-adversarial training, and its unrealized threats [[Paper](https://openreview.net/pdf?id=4t9q35BxGr), [Code](examples/attacks/inequality)]
 - **[ICASSP 2023]** TransAudio: Towards the Transferable Adversarial Audio Attack via Learning Contextualized Perturbations [[Paper](https://arxiv.org/abs/2303.15940), [Code](examples/attacks/transaudio)]
 - **[TIP 2022]** Towards Understanding and Boosting Adversarial Transferability from a Distribution Perspective [[Paper](https://arxiv.org/abs/2210.04213), [Code](examples/attacks/dra)]

examples/imageclassification/imagenet/dat/main.py

@@ -39,6 +39,7 @@
 from easyrobust.datasets import ImageNetDataset
 from easyrobust.third_party.vqgan import VQModel, reconstruct_with_vqgan
 from easyrobust.attacks import pgd_generator
+import easyrobust.models
 
 try:
     from apex import amp