![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 85
骑走的小木马
这个作者很懒,什么都没留下…
展开
-
pytorch训练模型时,修改代码后报RuntimeError: Trying to backward through the graph a second time
我在网上也查了相关的问题,各个博主也给了出现这个错误的理由以及要解决这个错误的方法,总结了一下,①:是在调用bachward()或者autograd.grad()函数时,指定retain_graph = True,意思就是把梯度值保存了;#修改前#修改后没有效果,训练使用的torch版本是1.10的,我看里面的代码,torch的_torsor.py文件里的backward()函数确实有该参数,但是我加上了,一点用都没有,还是报一样的错误,有点奇怪了。原创 2024-02-06 15:41:29 · 3146 阅读 · 0 评论 -
使用django 加载pytorch训练好的模型
近期使用PointRend模型来做项目上目标分割部分,整个项目也都完成了,现在需要进行落地,让我有点烦恼,因为一期的项目都是基于tensorflow框架来加载所有的模型,现在使用的是pytorch框架,而且现在的项目中也使用到了tensorflow模型框架,虽然后期可以改成统一使用pytorch框架下训练的模型,但是现在需要呈现出效果,所以还是想可以快速实现。原先项目都是基于django做web服务器落地提供API接口给前端访问,所以我这次也是准备这么做,但是我面临两个问题: ①:django要...原创 2020-08-15 10:45:58 · 8404 阅读 · 15 评论