fastNLP.modules.utils

fastNLP.modules.utils.initial_parameter(net, initial_method=None)[源代码]

A method used to initialize the weights of PyTorch models.

参数

  • net -- a PyTorch model

  • initial_method (str) --

    one of the following initializations.

    • xavier_uniform

    • xavier_normal (default)

    • kaiming_normal, or msra

    • kaiming_uniform

    • orthogonal

    • sparse

    • normal

    • uniform

fastNLP.modules.utils.summary(model: torch.nn.modules.module.Module)[源代码]

别名 fastNLP.modules.summary fastNLP.modules.utils.summary

得到模型的总参数量

Params model

Pytorch 模型

Return tuple

包含总参数量，可训练参数量，不可训练参数量