composer.models.initializers#

Module 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.

get_initializer()[source]#

Get the initializer function.

Returns

(torch.nn.Module) -> None โ€“ The initializer function.