WindVChen / DiffAttack

An unrestricted attack based on diffusion models that can achieve both good transferability and imperceptibility.
Apache License 2.0
115 stars 9 forks source link

I want to train this model #10

Open nil0330 opened 9 months ago

nil0330 commented 9 months ago

I want to train your model, but I couldn't find the corresponding train.py. How can I train this model?

WindVChen commented 9 months ago

Hi @nil0330 ,

Our method is optimization-based and doesn't require training using train.py. Please refer to here for optimization commands and parameters. You can also find more details in diff_latent_attack.py.

nil0330 commented 9 months ago

Thanks for your answer !