2023年7月21日
with torch.no_grad():注意事项
摘要： 1。 当执行原地操作时，例如 tensor.add_(x)，将会在一个张量上直接修改数据，而不会创建新的张量。由于修改了张量的数据，因此计算图会失效，即计算图中的操作和输入输出关系都会发生变化。这会导致反向传播无法正确计算梯度。因此，PyTorch 禁止在需要梯度计算的张量上执行原地操作。为了解决这 阅读全文
posted @ 2023-07-21 10:21 ML_WG 阅读(461) 评论(0) 推荐(0)