parameters
和children
似乎显示相同的信息,所以它们之间有什么区别?
import torch
print('torch.__version__', torch.__version__)
m = torch.load('imagenet_resnet18.pth')
print(m.parameters)
print(m.children)
答案 0 :(得分:0)
model.parameters()
是一个生成器,它返回包含您的模型参数的张量。model.children()
是一个生成器,可返回模型的各层,您可以使用<layername>.weight
或<layername>.bias
访问此link,获取有关访问和冻结模型层的简单教程。