Я хочу, чтобы некоторые из моих моделей были заморожены. После официальных документов:
with torch.no_grad():
linear = nn.Linear(1, 1)
linear.eval()
print(linear.weight.requires_grad)
Но он печатает True
вместо False
. Если я хочу установить модель в режиме eval, что мне делать?