composer.models.initializers#

composer.models.initializers

Classes

Initializer

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

StringEnum

Base class for Enums containing string values.

Attributes

  • Callable

class composer.models.initializers.Initializer(value)[source]#

Bases: composer.utils.string_enum.StringEnum

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