weight_initializer

  • type: xavier normal (default) / xavier uniform, normal, constant, kaiming normal, kaiming uniform, he normal, he uniform, orthogonal, sparse


Weight initialization is used to define the initial values for the parameters in Neural Network models prior to training the models on dataset. SIMPLE-NN supports several weight_initializer and the usage of weight_initializer is as below.

# input.yaml
neural_network:
    weight_initializer:
        type: xavier normal
        params: