指定网络不反向传播给定函数中的梯度

Question

在我的神经网络中，为了计算损失，我需要在训练的时候做一些中间计算，首先得到一些变换rv。

rv = factor.ransac(source, target, prob, device)
predicted = factor.predict(source, rv, outputs, device)
loss = criterion(predicted, target)

我只想通过predicted 而不是通过factor.ransac 反向传播渐变。我该怎么做？

Answer 1

您可以使用detach()。

rv = factor.ransac(source, target, prob, device).detach()

这样，rv 上的渐变将被丢弃。