RuntimeError:a leaf Variable that requires grad is being used is an in-place operation.
Aria
下载可以很简单
项目地址:https://gitcode.com/gh_mirrors/ar/Aria
免费下载资源
·
没有加with torch.no_grad()
with torch.no_grad()
对于推断非常有用,如果确定不会调用,它将减少计算的内存消耗,否则会有requires_grad=True。
在这种模式下,每次计算的结果都将具有requires_grad=False,即使输入具有requires_grad=True。
GitHub 加速计划 / ar / Aria
5.52 K
861
下载
下载可以很简单
最近提交(Master分支:3 个月前 )
b0d3c6dd - 5 个月前
8fd9634d - 5 个月前
更多推荐
已为社区贡献7条内容
所有评论(0)