Pytorch: заморозить слой и сделать градиенты слоя равными 0.Python

Программы на 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). Однако градиентов не ожидается.

Вернуться в «Python»