modify image representations to expect parameter containers instead of tensors.

pytti-tools / pytti-core

https://pytti-tools.github.io/pytti-book/intro.html

MIT License

78 stars 23 forks source link

modify image representations to expect parameter containers instead of tensors. #193

Open dmarx opened 2 years ago

dmarx commented 2 years ago

see https://github.com/pytti-tools/pytti-core/pull/192#issuecomment-1144303103

dmarx commented 2 years ago

I think the better solution here is to add EMA as a submodule on the image models that need it using ema-pytorch and just deprecate the EMAImage class entirely