В чем разница между параметрами и детьми? - PullRequest
0 голосов
/ 23 сентября 2018

Похоже, parameters и children показывают одинаковую информацию, так в чем же разница между ними?

import torch

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

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

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

1 Ответ

0 голосов
/ 09 октября 2018
  • model.parameters() - это генератор, который возвращает тензоры, содержащие параметры вашей модели.
  • model.children() - это генератор, который возвращает слои модели, из которых вы можете извлечь свои тензоры параметров, используя <layername>.weightили <layername>.bias

Посетите эту ссылку , чтобы получить простое руководство по доступу и заморозке слоев модели.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...