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

[FEATURE]: Add to Registry from Configs #29

Closed nmichlo closed 2 years ago

nmichlo commented 2 years ago

Is your feature request related to a problem? Please describe. Currently need to manually call and add items to disent.registry

Describe the solution you'd like Add support for experiment/run.py such that it can read from the config to add items to the registry for use.

nmichlo commented 2 years ago

Basic support added with custom initialiser functions.