DDP不能直接传入需要梯度的参数,而是要把不需要梯度的参数requirs_grad设置为False,模型所有参数都传入优化器!!
【无标题】
最新推荐文章于 2024-09-03 17:37:40 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)
DDP不能直接传入需要梯度的参数,而是要把不需要梯度的参数requirs_grad设置为False,模型所有参数都传入优化器!!