composer.models.model_hparams#

General YAHP interface for Base ComposerModels.

Hparams

These classes are used with yahp for YAML-based configuration.

ModelHparams

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 a ComposerModel.

Returns

ComposerModel โ€“ The constructed ComposerModel