如果你使用了in-place operation而没有报错的话,那么你可以确定你的梯度计算是正确的。 补充知识:PyTorch中nn.ReLU(inplace=True)中inplace的作用.
確定! 回上一頁