hy2dl.modelzoo

hy2dl.modelzoo.get_head(cfg: Config) → Module

Get head object, depending on the run configuration.

Parameters:: cfg (Config) – The run configuration.
Returns:: A new head instance of the type specified in the config.
Return type:: nn.Module

hy2dl.modelzoo.get_head_registry()

hy2dl.modelzoo.get_model(cfg: Config) → Module

Get model object, depending on the run configuration.

Parameters:: cfg (Config) – The run configuration.
Returns:: A new model instance of the type specified in the config.
Return type:: nn.Module

hy2dl.modelzoo.get_model_registry()