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