使用pytorch的loss.backward()时,出现element 0 of tensors does not require grad and does not have a grad_fn
element
A Vue.js 2.0 UI Toolkit for Web
项目地址:https://gitcode.com/gh_mirrors/eleme/element
免费下载资源
·
仅作为记录,大佬请跳过。
用loss.requires_grad_(True)
是因为loss没有设置梯度(所以不能反向传播)(loss的数据类型是tensor)
设置梯度后即可,展示:
参考
pytorch官方教程 传送门
GitHub 加速计划 / eleme / element
54.06 K
14.63 K
下载
A Vue.js 2.0 UI Toolkit for Web
最近提交(Master分支:3 个月前 )
c345bb45
7 个月前
a07f3a59
* Update transition.md
* Update table.md
* Update transition.md
* Update table.md
* Update transition.md
* Update table.md
* Update table.md
* Update transition.md
* Update popover.md 7 个月前
更多推荐
已为社区贡献9条内容
所有评论(0)