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:包含总参数量，可训练参数量，不可训练参数量