RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn

 

 

 

 报错如下:

 

原因: 这里的loss默认的requires_grad是False,因此在backward()处不会计算梯度,导致出错

方法:将loss的requires_grad属性设为True,如下所示:

 

posted @ 2023-02-06 18:51  呦呦南山  阅读(5526)  评论(0编辑  收藏  举报