特殊的bug:element 0 of tensors does not require grad and does not have a grad_fn

很多帖子都说了,设置requires_grad_()就行。

但是我这次遇到的不一样,设置了都不行。

我是这种情况,在前面设置了torch.no_grad():,又在这个的作用域下进行了requires_grad_(),这是不起作用的。

简单版:

py 复制代码
with torch.no_grad():
            model.eval()
            pos_embed = model(homo_data.x, homo_data.edge_index)
						....            
            pos_embed.requires_grad_()# 不起作用
            ....

这样子直接看,傻瓜都不会犯错。而我这个就比较隐蔽了。。。

我的:

py 复制代码
# file1.py
with torch.no_grad():
    pos_embed = model(homo_data.x, homo_data.edge_index)
    ...
    fun_A(pos_embed)
    ...
# file2.py
def fun_A(x):
    ...
    fun_B(x)
    ...
# file3.py
def fun_B(x):
	...
  x.requires_grad_() # 由于x是在torch.no_grad中传过来的,所以不起作用
  ...

就这样一直会报错element 0 of tensors does not require grad and does not have a grad_fn

太坑了

大家引以为鉴,看看出现这个情况没。

相关推荐
山烛10 分钟前
OpenCV图像形态学操作
图像处理·人工智能·python·opencv·计算机视觉·图像形态学
Q_Q51100828518 分钟前
python的校园研招网系统
开发语言·spring boot·python·django·flask·node.js·php
AI_RSER1 小时前
遥感&机器学习入门实战教程|Sklearn 案例④ :多分类器对比(SVM / RF / kNN / Logistic...)
python·算法·机器学习·支持向量机·分类·sklearn
于越海1 小时前
Python工程师向项目管理转型的深度分析与学习道路规划
笔记·python·学习
二闹2 小时前
discard和remove的那些事儿
python
大志说编程2 小时前
LangChain框架入门18: 十分钟带你搞定LLM工具调用
python·langchain·ai编程
钢铁男儿2 小时前
使用 TensorBoardX 实现 PyTorch 神经网络可视化:从入门到进阶
人工智能·pytorch·神经网络
AI视觉网奇4 小时前
麒麟系统播放图片 速度比较
开发语言·python·pygame
言之。4 小时前
Django REST框架核心:GenericAPIView详解
数据库·python·django
这里有鱼汤4 小时前
别傻了,这些量化策略AI 10 秒就能帮你写好
后端·python