使用pytorch中nn.ReLU(inplace=True)
因为inplace op会覆盖某些函数计算渐变所需的一些数据。覆盖它后,它无法做到这一点。只需inplace=True在这些地方删除。
RuntimeError: one of the variables needed for gradient computation has been modified by an inplace
最新推荐文章于 2024-06-28 18:52:56 发布
使用pytorch中nn.ReLU(inplace=True)
因为inplace op会覆盖某些函数计算渐变所需的一些数据。覆盖它后,它无法做到这一点。只需inplace=True在这些地方删除。