search

pytti-tools / pytti-core

https://pytti-tools.github.io/pytti-book/intro.html
MIT License
78 stars 23 forks source link

Migrate EMA backend to lucidrains' ema-pytorch #209

Open dmarx opened 2 years ago

dmarx commented 2 years ago

https://github.com/lucidrains/ema-pytorch/blob/main/ema_pytorch/ema_pytorch.py

Current EMAImage assumes we only need to track EMA on a single tensor, which would preclude EMA-ing a DIP image model (or any other implicit model.