首页 Paddle框架 帖子详情
运行alphafold_paddle/model/module.py报错
收藏
快速回复
Paddle框架 文章炼丹技巧模型训练深度学习 62 0
运行alphafold_paddle/model/module.py报错
收藏
快速回复
Paddle框架 文章炼丹技巧模型训练深度学习 62 0

python==3.7.0

paddlepaddle-gpu==2.5.1

【报错】

File "/root/autodl-tmp/helixfold-single/alphafold_paddle/model/modules.py", line 485,in forward q_data, m_data,self.query_w, self.key_w, self.value_w, self.qkv_w, nonbatched_bias, bias, self.gating_w,self.gating_b,self.output_w, self.output_b,' has_gating', self.config.gating, ' merge  qkv', self.merge_qkv)
ValueError: too many values to unpack (expected 8)

【解决方法】

定位到源代码

 _, _, _, _, _, _, _, output = _C_ops.fused_gate_attention(q_data, m_data, self.query_w, self.key_w, self.value_w, self.qkv_w, nonbatched_bias, bias, self.gating_w, self.gating_b,self.output_w, self.output_b, 'has_gating', self.config.gating, 'merge_qkv', self.merge_qkv)

由于_C_ops.fused_gate_attention应该返回9个元素,此处只有8个元素,个数不匹配,故报错,修改上述代码如下即可:

 _, _, _, _, _, _, _, _, output = _C_ops.fused_gate_attention(q_data, m_data, self.query_w, self.key_w, self.value_w, self.qkv_w, nonbatched_bias, bias, self.gating_w, self.gating_b,self.output_w, self.output_b, 'has_gating', self.config.gating, 'merge_qkv', self.merge_qkv)
0
收藏
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户