search

htyjers / StrDiffusion

[CVPR 2024] Structure Matters: Tackling the Semantic Discrepancy in Diffusion Models for Image Inpainting
Apache License 2.0
85 stars 11 forks source link

train epoch #7

Closed tianxiaoer closed 6 months ago

tianxiaoer commented 6 months ago

Hello, I would like to ask, during training, how many epochs do you recommend training for?

htyjers commented 6 months ago

For small dataset, I think you can set the 'niter: 700000' and '_lrsteps: [200000, 400000, 600000]'，and the big dataset can set more niter for trainning