search

JuliaWolleb / Diffusion-based-Segmentation

This is the official Pytorch implementation of the paper "Diffusion Models for Implicit Image Segmentation Ensembles".
MIT License
280 stars 38 forks source link

EPOCH, and time steps #41

Open JigneshChowdary opened 1 year ago

JigneshChowdary commented 1 year 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 1 year ago

they run 60000 iterations, mentioned in paper

JuliaWolleb commented 1 year 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.