SCLBD/BackdoorBench

RuntimeError: DataLoader worker (pid(s) 35464) exited unexpectedly

Closed this issue · 1 comments

Hello, thank you for the thorough summary you provided. I encountered a problem while executing a command; when i executing

python ./attack/badnet.py --yaml_path ../config/attack/prototype/cifar10.yaml --patch_mask_path ../resource/badnet/trigger_image.png

image

there might be an infinite loop in the getattr method within bd_dataset_v2.py. I searched online for a solution and found that reducing the number of num_workers could help. I have already adjusted it to 1, but the problem persists. Could you suggest any other possible solutions?
Thanks!

python ./attack/badnet.py --yaml_path ../config/attack/prototype/cifar10.yaml --patch_mask_path ../resource/badnet/trigger_image.png

set num_workers = 0 can solve the problem