Custom dataset

[tfc-0]

Hi, I want to train this for a custom dataset for my research purpose? Can you tell me which scripts I need to run for training a scratch model after modifying dataloader?

[valentyn1boreiko]

Hi, thank you for you interest!

Given that you have already your (non-robust) classifier that you would like to explain, you only need to train on your dataset:

• an adversarially robust classifier using for example robustness package and optionally robust-finetuning package,
• a diffusion model using guided-diffusion package.

Do you have any further questions?

[ShruthiVijayakumar]

for how many steps/epochs you trained the diffusion model and non-robust model for imagenet dataset?