Closed arhanjain closed 1 week ago
Thanks! I think it'd be much better to move this to EncoderCore in obs_core.py, as many classes that inherits Module
are not necessarily encoders
looks like there's already a encoder registry, so we just need to modify the way the VisualCore class initializes its backbone!
Thanks @arhanjain for your contribution!
Addresses #189
The implementation consists of adding a registry to the Module class so that all child classes are registered by their name into the registry upon
__init_subclass__
. This allows one to initialized backbones (like in parameters of VisualCore) through the registry rather than usingeval
which only allows models from base_nets.pyExample script coming soon