参数和儿童有什么区别？

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

访问此链接以获取有关访问和冻结模型图层的简单教程。