composer.models.model_hparams#
General YAHP interface for Base ComposerModels.
Hparams
These classes are used with yahp
for YAML
-based configuration.
General YAHP interface for ComposerModels. |
- class composer.models.model_hparams.ModelHparams(initializers=<factory>, num_classes=None)[source]#
Bases:
yahp.hparams.Hparams
,abc.ABC
General YAHP interface for ComposerModels.
- Parameters
num_classes (int) โ The number of classes. Needed for classification tasks. Default:
None
.initializers (List[Initializer], optional) โ The initialization strategy for the model. Default:
None
.
- abstract initialize_object()[source]#
Invoked by the
initialize_object()
to construct aComposerModel
.- Returns
ComposerModel โ The constructed
ComposerModel