工具

梯度裁剪

参数初始化

initial_parameter(net, initial_method=None)[源代码]

用于初始化 Pytorch 模型内部权重的函数

参数

  • net – Pytorch Module

  • initial_method (str) –

    取值如下

    • xavier_uniform

    • xavier_normal (default)

    • kaiming_normal, or msra

    • kaiming_uniform

    • orthogonal

    • sparse

    • normal

    • uniform