composer.models.model_hparams#

General YAHP interface for ComposerModels.

Classes

Initializer

Sets the initialization scheme for different layers of a PyTorch model.

Hparams

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

ModelHparams

General YAHP interface for ComposerModels.
class composer.models.model_hparams.Initializer(value)[source]#

Bases: composer.utils.string_enum.StringEnum

Sets the initialization scheme for different layers of a PyTorch model.

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