Код: Выделить всё
import torch
layer = torch.nn.Linear(10, 10)
layer.weight = torch.nn.Parameter(torch.ones_like(layer.weight))
Может ли кто-нибудь объяснить мне, почему контекстный менеджер здесь необходим, несмотря на то, что внутри него не вычисляются градиенты, если я делаю что-то вроде
Код: Выделить всё
with torch.no_grad():
layer.weight = torch.nn.Parameter(torch.ones_like(layer.weight))
Я пытался немного покопаться в документации PyTorch, чтобы понять это, но не получил где угодно.
Подробнее здесь: https://stackoverflow.com/questions/789 ... orch-layer