search

JuliaWolleb / Diffusion-based-Segmentation

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

emasavedmodel、optsavedmodel、savedmodel #42

Open yuan5828225 opened 1 year ago

yuan5828225 commented 1 year ago

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

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