Hyperparameter values for ControlNet with semantic segmentation

Open Dan-Leh opened this issue 3 years ago • 3 comments

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!

Feb 20 '23 14:02 Dan-Leh

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

Feb 23 '23 12:02 Dan-Leh

I'm also interested in this.

Feb 28 '23 10:02 omaralvarez

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.

Mar 19 '23 10:03 Sierkinhane

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

Jul 04 '23 07:07 hxy-123-coder