hamadichihaoui / BIRD

This is the official implementation of "Blind Image Restoration via Fast Diffusion Inversion"
243 stars 18 forks source link

How to train my own datasets? #9

Open superY688 opened 4 months ago

superY688 commented 4 months ago

Hello, I wanted to express my gratitude for your work; it's been instrumental in my current project. However, I've encountered some confusion that I'm hoping you can shed light on. Now,I want to train my own datasets.However,i can not find the related code.Can you provide the code or instruction which i can use to train my datasets. Thank you very much.

hamadichihaoui commented 4 months ago

Hi @superY688 and thanks for your interest in our work. Here, we used a pre-trained model. If you want to train your own diffusion model using a custom datasets, you can use OpenAI repo, https://github.com/openai/improved-diffusion.git