search

xavihart / Diff-PGD

[NeurIPS'2023] Official Code Repo:Diffusion-Based Adversarial Sample Generation for Improved Stealthiness and Controllability
87 stars 7 forks source link

env.yml seems to be incorrect #3

Closed SilverRAN closed 1 year ago

SilverRAN commented 1 year ago

There is an error when running attack_global.py:

Traceback (most recent call last): File "code/attack_global.py", line 1, in from load_dm import get_imagenet_dm_conf File "/home/XXX/Diff-PGD/code/load_dm.py", line 23, in from utils import * File "/home/XXX/Diff-PGD/code/utils.py", line 4, in from colorama import Fore, Back, Style ModuleNotFoundError: No module named 'colorama'

I tried to install colorama using conda, however, another error occurred:

Traceback (most recent call last): File "code/attack_global.py", line 8, in from archs import get_archs, IMAGENET_MODEL File "/home/XXX/Diff-PGD/code/archs.py", line 11, in from robustbench.utils import load_model File "/home/XXX/anaconda3/envs/diff-pgd/lib/python3.8/site-packages/robustbench/init.py", line 1, in from .data import load_cifar10 File "/home/XXX/anaconda3/envs/diff-pgd/lib/python3.8/site-packages/robustbench/data.py", line 15, in from robustbench.model_zoo import model_dicts as all_models File "/home/XXX/anaconda3/envs/diff-pgd/lib/python3.8/site-packages/robustbench/model_zoo/init.py", line 1, in from .models import model_dicts File "/home/XXX/anaconda3/envs/diff-pgd/lib/python3.8/site-packages/robustbench/model_zoo/models.py", line 4, in from robustbench.model_zoo.cifar10 import cifar_10_models File "/home/XXX/anaconda3/envs/diff-pgd/lib/python3.8/site-packages/robustbench/model_zoo/cifar10.py", line 24, in from robustbench.model_zoo.architectures.robustarch_wide_resnet import get_model as get_robustarch_model File "/home/XXX/anaconda3/envs/diff-pgd/lib/python3.8/site-packages/robustbench/model_zoo/architectures/robustarch_wide_resnet.py", line 18, in from torchvision.ops.misc import Conv2dNormActivation, SqueezeExcitation ImportError: cannot import name 'Conv2dNormActivation' from 'torchvision.ops.misc' (/home/XXX/anaconda3/envs/diff-pgd/lib/python3.8/site-packages/torchvision/ops/misc.py)

xavihart commented 1 year ago

Thanks for raising this question. I have updated the env.yml with a stable version of robustbench, you can try again. If you have further questions, feel free to contact me!

SilverRAN commented 1 year ago

Everything is going well now, thanks!