How can I fine-tune the pretrained model on my single class dataset?

openai / guided-diffusion

MIT License

5.9k stars 782 forks source link

How can I fine-tune the pretrained model on my single class dataset? #143

Open Ucarrot319 opened 3 months ago

Ucarrot319 commented 3 months ago

I want to use the pre-trained model published by the author to fine-tune my target dataset, which is on a single scenario. Does anyone know how to do that, thank you