How to train checkpoint on own dataset

xavihart / Diff-PGD

[NeurIPS'2023] Official Code Repo:Diffusion-Based Adversarial Sample Generation for Improved Stealthiness and Controllability

84 stars 7 forks source link

Open wlf121 opened 2 weeks ago

wlf121 commented 2 weeks ago

Thank you for open-sourcing your wonderful work.How to train checkpoint on own dataset。

xavihart commented 2 weeks ago

please refer to https://github.com/openai/guided-diffusion for training a guided diffusion on your own dataset