[paddle 参数更多]torch.nn.Module.named_modules¶

torch.nn.Module.named_modules ¶

          torch.nn.Module.named_modules(memo=None, prefix='', remove_duplicate=True)

         

          paddle.nn.Layer.named_sublayers(prefix='', include_self=False, layers_set=None, remove_duplicate=True)

         

Paddle 相比 PyTorch 支持更多其他参数，具体如下：

PyTorch	PaddlePaddle	备注
memo	layers_set	用来记录已经加入结果的子层的集合，仅参数名不一致。
prefix	prefix	在所有参数名称前加的前缀。
remove_duplicate	remove_duplicate	是否删除结果中重复的模块实例。
-	include_self	是否包含该层自身，PyTorch 无此参数，Paddle 需设为 True 才与 Pytorch 一致。