search

sun-umn / DMPlug

This is the official implementation of "DMPlug: A Plug-in Method for Solving Inverse Problems with Diffusion Models" (NeurIPS 2024).
42 stars 3 forks source link

about ES–WMV #4

Open Jiangjiawei2 opened 1 week ago

Jiangjiawei2 commented 1 week ago

Hi! Nice work! I wonder which part of the code provides the ES–WMV. Currently, I need to run through all the iterations

HengkangWang commented 1 week ago

Hi @Jiangjiawei2 , thank you for your interest in our work! For further details on ES-WMV and the code, please visit the repository at https://github.com/sun-umn/Early_Stopping_for_DIP. Since ES-WMV is not the main innovation of this paper, we have kept it separate; however, it’s straightforward to use. You only need to save intermediate reconstructions, calculate the variance, and decide on an optimal stopping point. We encourage you to try it out, and if you encounter any challenges, we may consider releasing additional code to demonstrate how to integrate our method with ES-WMV.