search

lllyasviel / ControlNet

Let us control diffusion models!
Apache License 2.0
30.72k stars 2.76k forks source link

Hyperparameter values for ControlNet with semantic segmentation #118

Open Dan-Leh opened 1 year ago

Dan-Leh commented 1 year ago

Hi,

I wanted to ask what learning rate and batch size were used to train ControlNet with semantic segmentation for both the ADE20k and COCO datasets?

Thanks in advance!

Dan-Leh commented 1 year ago

@lllyasviel could you also add how many training iterations you used, in addition to learning rate and batch size as asked above?

omaralvarez commented 1 year ago

I'm also interested in this.

Sierkinhane commented 1 year ago

Is the model trained using MS-COCO available? It seems that [control_sd15_seg.pth](https://huggingface.co/lllyasviel/ControlNet/blob/main/models/control_sd15_seg.pth) is the model trained using ADE20K.

hxy-123-coder commented 1 year ago

@Sierkinhane @Dan-Leh Hello！Did you train ControlNet with ADE20K ? How the dataset should be organized if using ADE20k?