search

nmichlo / disent

🧶 Modular VAE disentanglement framework for python built with PyTorch Lightning ▸ Including metrics and datasets ▸ With strongly supervised, weakly supervised and unsupervised methods ▸ Easily configured and run with Hydra config ▸ Inspired by disentanglement_lib
https://disent.michlo.dev
MIT License
124 stars 18 forks source link

[CHANGE]: Remove Loss Reduction "sum" #12

Closed nmichlo closed 3 years ago

nmichlo commented 3 years ago

Consider removing additional loss reduction modes such as mean and standardise on the official VAE mode mean_sum

nmichlo commented 3 years ago

removed reduction='sum' in 7f7d757133bc483e4637be8ee57b86b0bc22e116 release v0.1.0

modes "mean_sum" and "mean" are still available but this still requires considerable effort to maintain and verify across models.