Make sure the zeros tensor for gradient accumulation for a loop invariant is...
Make sure the zeros tensor for gradient accumulation for a loop invariant is initialized with the right shape. This is needed because this zeros tensor can get passed downstream when the loop exits with no iterations. Change: 125620906
Showing
想要评论请 注册 或 登录