相关疑难解决方法(0)

pytorch如何设置.requires_grad错误

我想把我的一些模型冻结.遵循官方文档:

with torch.no_grad():
    linear = nn.Linear(1, 1)
    linear.eval()
    print(linear.weight.requires_grad)
Run Code Online (Sandbox Code Playgroud)

但它打印True而不是False.如果我想在eval模式下设置模型,我该怎么办?

python gradient-descent pytorch

17
推荐指数
3
解决办法
2万
查看次数

标签 统计

gradient-descent ×1

python ×1

pytorch ×1