Diffusion-based-Segmentation icon indicating copy to clipboard operation
Diffusion-based-Segmentation copied to clipboard

EPOCH, and time steps

Open JigneshChowdary opened this issue 1 year ago • 2 comments

Hi, Can I know for how many epochs, and time-steps the model is trianed for? Can I know how to change these?

JigneshChowdary avatar Jul 16 '23 20:07 JigneshChowdary

they run 60000 iterations, mentioned in paper

li-hai666 avatar Jul 18 '23 08:07 li-hai666

By default there is no stopping criterion. As mentioned above, we stopped after 60'000 iterations. If you want to implement a stopping criterion, adapt the code here.

JuliaWolleb avatar Oct 27 '23 09:10 JuliaWolleb