Похоже, parameters и children показывают одинаковую информацию, так в чем же разница между ними?
parameters
children
import torch print('torch.__version__', torch.__version__) m = torch.load('imagenet_resnet18.pth') print(m.parameters) print(m.children)
model.parameters()
model.children()
<layername>.weight
<layername>.bias
Посетите эту ссылку , чтобы получить простое руководство по доступу и заморозке слоев модели.