Custom dataset - Githubissues

valentyn1boreiko / DVCEs

35 stars 6 forks source link

Custom dataset #4

Open tfc-0 opened 1 year ago

tfc-0 commented 1 year ago

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 commented 1 year ago

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 commented 5 months ago

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