pytorch - 参数和 child 有什么区别？

Question

它看起来像 parameters和 children显示相同的信息，那么它们之间有什么区别？

import torch

print('torch.__version__', torch.__version__)

m = torch.load('imagenet_resnet18.pth')

print(m.parameters)
print(m.children)

Answer 1

model.parameters()是一个生成器，它返回包含模型参数的张量。

model.children()是一个返回模型层的生成器，您可以使用 <layername>.weight 从中提取参数张量或 <layername>.bias

访问此 link有关访问和卡住模型层的简单教程。