Skip to content

get_init_fn

source

.get_init_fn(
   method: str = 'orthogonal'
)


Returns a network initialization function.

Args

  • method (str) : Initialization method name.

Returns

Initialization function.


_xavier_normal

source

._xavier_normal(
   m
)


Xavier normal initialization.


_xavier_uniform

source

._xavier_uniform(
   m
)


Xavier uniform initialization.


_orthogonal

source

._orthogonal(
   m
)


Orthogonal initialization.


_identity

source

._identity(
   m
)


Identity initialization.