fastNLP.modules.utils module

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

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