modify image representations to expect parameter containers instead of tensors.

[dmarx]

see #192 (comment)

[dmarx]

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