Initializer#

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

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.