DBA icon indicating copy to clipboard operation
DBA copied to clipboard

Published 20 hours ago • verified publisher icon AI-secure

thanks for your code, I want to do this experiment on 256*256 datasets for example Imagenet, what can I do to make it?

Open JobsWangHaoQuan opened this issue 1 year ago • 1 comments

JobsWangHaoQuan avatar Apr 15 '23 13:04 JobsWangHaoQuan

Hi, for new dataset, you may want to

  1. add a new dataloader https://github.com/AI-secure/DBA/blob/3cabf7ec3d9967eefb9c3cb983c6eb671ac12b33/image_helper.py#L173
  2. add the dataset in the main.py https://github.com/AI-secure/DBA/blob/3cabf7ec3d9967eefb9c3cb983c6eb671ac12b33/main.py#L106
  3. parse the new parameters (https://github.com/AI-secure/DBA/blob/master/utils/tiny_params.yaml) for this dataset

AlphaPav avatar Apr 21 '23 16:04 AlphaPav