EPOCH, and time steps - Githubissues

JuliaWolleb / Diffusion-based-Segmentation

This is the official Pytorch implementation of the paper "Diffusion Models for Implicit Image Segmentation Ensembles".

MIT License

271 stars 35 forks source link

EPOCH, and time steps #41

Open JigneshChowdary opened 12 months ago

JigneshChowdary commented 12 months ago

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

li-hai666 commented 11 months ago

they run 60000 iterations, mentioned in paper

JuliaWolleb commented 8 months ago

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.