No module named 'ema_pytorch'

Algolzw / image-restoration-sde

Image Restoration with Mean-Reverting Stochastic Differential Equations, ICML 2023. Winning solution of the NTIRE 2023 Image Shadow Removal Challenge.

https://algolzw.github.io/ir-sde/index.html

MIT License

521 stars 39 forks source link

No module named 'ema_pytorch' #60

Closed tan0427 closed 8 months ago

tan0427 commented 8 months ago

作者你好，我在复现这篇论文的时候，报错No module named 'ema_pytorch'，我找了一些文件，都没有找到这个包，请问这个包在哪里呢？

ygtxr1997 commented 7 months ago

应该是这个：https://github.com/lucidrains/ema-pytorch