LightningCLI custom registry #12782
Unanswered
udeepam
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 1 comment
-
There is no need for registries to make a parameter (with type hint a class) configurable from the config file. In a config file you specify the full |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
Is there a way to add a custom registry to the LightningCLI?
Currently the LightningCLI instantiates classes from the following registries: https://github.com/PyTorchLightning/pytorch-lightning/blob/master/pytorch_lightning/utilities/cli.py#L107
What I would like to do is make a
CRITERION_REGISTRY
so that I can make custom criteria and make the criteria configurable from the config file. This would then be passed to the model.Beta Was this translation helpful? Give feedback.
All reactions