分类：model.eval()

2021
07-09

聊聊pytorch测试的时候为何要加上model.eval()

Doneedtousemodel.eval()whenItest?Sure,Dropoutworksasaregularizationforpreventingoverfittingduringtraining.ItrandomlyzerostheelementsofinputsinDropoutlayeronforwardcall.Itshouldbedisabledduringtestingsinceyoumaywanttousefullmodel(noelementismasked)使用PyTorch进行训练和测试时一定注意要把实例化的model指定train/eval，eval（）时，框架会自动把BN和DropOut... 继续阅读 >

2021
07-09

pytorch中的model.eval()和BN层的使用

看代码吧~classConvNet(nn.module):def__init__(self,num_class=10):super(ConvNet,self).__init__()self.layer1=nn.Sequential(nn.Conv2d(1,16,kernel_size=5,stride=1,padding=2),nn.BatchNorm2d(16),nn.ReLU(),nn.MaxPool2d(kernel_size=2,stride=2))self.layer2=nn.S... 继续阅读 >