cabralpinto / modular-diffusion

Python library for designing and training your own Diffusion Models with PyTorch.
https://cabralpinto.github.io/modular-diffusion/
MIT License
266 stars 12 forks source link

Decoder parameters #17

Open jbgao opened 1 month ago

jbgao commented 1 month ago

Hi All

How shall we optimize the parameters in decoder? It seems the module only add encoder parameters to optimiser?

J.