search

aks2203 / poisoning-benchmark

A unified benchmark problem for data poisoning attacks
https://arxiv.org/abs/2006.12557
MIT License
146 stars 21 forks source link

Unable to generate poison dataset for testing #16

Closed ArshadIram closed 1 year ago

ArshadIram commented 1 year ago

I am trying to run poison_test file based on the provided poison example. However, I am getting error while creating a dataloader of the poison dataset.

image

image

aks2203 commented 1 year ago

Hi there, can you please provide a little more information so I can recreate the error. What version of python? What command (and with which args) led to this error?

ArshadIram commented 1 year ago

python poison_test.py --model resnet18 --model_path pretrained_models//ResNet18_CIFAR10_adv.pth --poisons_path bp_poisons//num_poisons=25//0 --dataset cifar10

python version: 3.10.7

ArshadIram commented 1 year ago

The issue is fixed. The error occurred because the trainset_size argument was not provided.