The wrapper for the GDP can be found in brew_poison.py. To craft poisons on CIFAR-10 and ResNet18 with the default values, you can use the following command:

There are a bunch of optional arguments in the forest/options.py. Here are some of them:

@misc{souri2024generating,
      title={Generating Potent Poisons and Backdoors from Scratch with Guided Diffusion},
      author={Hossein Souri and Arpit Bansal and Hamid Kazemi and Liam Fowl and Aniruddha Saha and Jonas Geiping and Andrew Gordon Wilson and Rama Chellappa and Tom Goldstein and Micah Goldblum},
      year={2024},
      eprint={2403.16365},
      archivePrefix={arXiv},
      primaryClass={cs.LG}
}