weixin_43844219的博客

原创 [pytorch]model.children(), model.modules()和parameters()的区别和用法

[pytorch]children,modules和parameters的区别和用法childrenmoduleparameterschildrenchildren只获取最浅层的网络结构，相应的named_children则返回tuple的数据，tuple[0]是该层的名称，tuple[1]是相应的结构：class Net4(torch.nn.Module): def __init__(self): super(Net4, self).__init__() se