Pytorch: заморозить слой и сделать градиенты слоя равными 0. ⇐ Python
-
Гость
Pytorch: заморозить слой и сделать градиенты слоя равными 0.
Как правильно сделать нейронную сеть pytorch ложной?
(1) В этом сообщении: https://www.cs.toronto.edu/~lczhang/321 ... notes.html они установили model.requires_grad=False. Однако когда я проверяю
для имени, параметра в model.named_parameters(): печать (имя, параметр.requires_grad) У меня все верно. И все параметры param.grad не равны 0.
(2) Если я установлю param.requires_grad=False, я получу для всех param.grad значение None
То же самое я пытаюсь сделать и со ссылкой в (1). Однако градиентов не ожидается.
Как правильно сделать нейронную сеть pytorch ложной?
(1) В этом сообщении: https://www.cs.toronto.edu/~lczhang/321 ... notes.html они установили model.requires_grad=False. Однако когда я проверяю
для имени, параметра в model.named_parameters(): печать (имя, параметр.requires_grad) У меня все верно. И все параметры param.grad не равны 0.
(2) Если я установлю param.requires_grad=False, я получу для всех param.grad значение None
То же самое я пытаюсь сделать и со ссылкой в (1). Однако градиентов не ожидается.
Мобильная версия