Diffusion-based-Segmentation
Diffusion-based-Segmentation copied to clipboard
EPOCH, and time steps
Hi, Can I know for how many epochs, and time-steps the model is trianed for? Can I know how to change these?
they run 60000 iterations, mentioned in paper
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.