2021
07-17
07-17
PyTorch dropout设置训练和测试模式的实现
看代码吧~classNet(nn.Module):…model=Net()…model.train()#把module设成训练模式，对Dropout和BatchNorm有影响model.eval()#把module设置为预测模式，对Dropout和BatchNorm模块有影响补充：Pytorch遇到的坑——训练模式和测试模式切换由于训练的时候Dropout和BN层起作用，每个batchBN层的参数不一样，dropout在训练时随机失效点具有随机性，所以训练和测试要区分开来。使用时切记要根据实际情况切换：model.train()mode...
继续阅读 >