nn.Dropout(inplace=True

根据pytorch>Docs > Autograd mechanics描述,inplace会用执行随机失活后的结果覆盖原来的输入,改变了存储值,但随机失活并不会影响梯度计算和反向传播。

你可能感兴趣的:(pytorch,pytorch)