动态图代码在paddle 1.5中可以正常训练,在paddle 1.6中经过一个step会抛出梯度维度不一致的异常
Created by: wwjjy
- 版本、环境信息: 1)PaddlePaddle版本:paddle 1.5、paddle 1.6 2)系统环境:Python 3.7
- 训练信息 1)单机 2)显存信息 3)Operator信息
- 复现信息:该问题详细代码以及复现方式可在aistudio中查看,项目地址为:https://aistudio.baidu.com/aistudio/projectdetail/201493 (目前为1.5版本,可fork后切换到1.6)
- 问题描述:上述项目在paddle 1.5的环境下可以正常训练,但是在paddle 1.6环境中经过一个step后会抛出异常