参数和子代之间有什么区别?

时间:2018-09-23 11:36:32

标签: pytorch

parameterschildren似乎显示相同的信息,所以它们之间有什么区别?

import torch

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

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

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

1 个答案:

答案 0 :(得分:0)

  • model.parameters()是一个生成器,它返回包含您的模型参数的张量。
  • model.children()是一个生成器,可返回模型的各层,您可以使用<layername>.weight<layername>.bias
  • 从中提取参数张量

访问此link,获取有关访问和冻结模型层的简单教程。