initial_parameter
用于初始化 Pytorch 模型内部权重的函数
net – Pytorch Module
initial_method (str) –
取值如下
xavier_uniform
xavier_normal (default)
kaiming_normal, or msra
kaiming_uniform
orthogonal
sparse
normal
uniform