search

lvpeizhuo / Data-free_Backdoor

This is the source code for Data-free Backdoor. Our paper is accepted by the 32nd USENIX Security Symposium (USENIX Security 2023).
MIT License
28 stars 2 forks source link

您好！代码poison_model.py无法运行 #5

Closed aosodofogo closed 2 weeks ago

aosodofogo commented 1 month ago

首先是我将第75行 if name == 'layer4.0.conv1.weight': 修改为 if name == 'module.layer4.0.conv1.weight': 这部分是保证模型lay4之后的保持更新。

然后，在utils.py的第163行，output在epoch等于1 的时候，output = model(input)输出结果全是nan，请问为什么会这样呢

aosodofogo commented 4 weeks ago

将预训练模型加载改成resnet50 解决了这个问题