search

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

emasavedmodel、optsavedmodel、savedmodel #42

Open yuan5828225 opened 11 months ago

yuan5828225 commented 11 months ago

In the last , i saved emasavedmodel、optsavedmodel、savedmodel , and what's the difference between them ?

JuliaWolleb commented 8 months ago

Hi

"ema" is the exponential moving average. "opt" is the optimizer state you need to resume the training from a checkpoint. For inference, I chose "savedmodel".