search

THUYimingLi / BackdoorBox

The open-sourced Python toolbox for backdoor attacks and defenses.
GNU General Public License v2.0
460 stars 73 forks source link

运行test_SCALE_UP.py遇到一点问题 #79

Closed lungyam closed 1 month ago

lungyam commented 1 month ago

您好，我运行 test_Wanet.py 获得了 Wanet 中毒模型，在运行 test_SCALE_UP.py 测试防御效果时，加载 wanet 模型遇到了 'netC' 键的缺失问题，打印查看模型键值对时，发现模型似乎没有 'netC' 这个键： 1 2

请问是我的处理方式有问题吗？

LandAndLand commented 1 month ago

您好，我运行 test_Wanet.py 获得了 Wanet 中毒模型，在运行 test_SCALE_UP.py 测试防御效果时，加载 wanet 模型遇到了 'netC' 键的缺失问题，打印查看模型键值对时，发现模型似乎没有 'netC' 这个键： 1 2

请问是我的处理方式有问题吗？

你好，这个属于是加载模型的问题。因为我在训练模型的时候把模型权重保存在‘netC’关键字中了，所以需要去“netC”里面找模型权重，而你运行test_wanet.py是直接保存模型权重，所以可以用以下方式来加载模型： model.load_state_dict(torch.load(model_path), strict=False)

lungyam commented 1 month ago

您好，我运行 test_Wanet.py 获得了 Wanet 中毒模型，在运行 test_SCALE_UP.py 测试防御效果时，加载 wanet 模型遇到了 'netC' 键的缺失问题，打印查看模型键值对时，发现模型似乎没有 'netC' 这个键： 1 2 请问是我的处理方式有问题吗？

你好，这个属于是加载模型的问题。因为我在训练模型的时候把模型权重保存在‘netC’关键字中了，所以需要去“netC”里面找模型权重，而你运行test_wanet.py是直接保存模型权重，所以可以用以下方式来加载模型： model.load_state_dict(torch.load(model_path), strict=False)

感谢您的解答！！