search

cyclomon / DiffuseIT

Official repository of "Diffusion-based Image Translation using Disentangled Style and Content Representation" ( ICLR 2023 )
MIT License
257 stars 15 forks source link

How to train a custom model #13

Closed OmSDeshmukh closed 6 months ago

OmSDeshmukh commented 6 months ago

Hey, I tried your pre-trained models. They are really good. Do you have code to train a custom model on a different dataset from the ones mentioned in the Readme file.

cyclomon commented 6 months ago

Hi, the pre-trained model I uploaded is brought from ADM model (Diffusion Models Beat GANs on Image Synthesis, https://github.com/openai/guided-diffusion ) . You can train the new model using the same protocol from ADM.

OmSDeshmukh commented 6 months ago

Thanks!