分类：pytorch测试

2021
07-17

pytorch 实现在测试的时候启用dropout

我们知道，dropout一般都在训练的时候使用，那么测试的时候如何也开启dropout呢？在pytorch中，网络有train和eval两种模式，在train模式下，dropout和batchnormalization会生效，而val模式下，dropout不生效，bn固定参数。想要在测试的时候使用dropout，可以把dropout单独设为train模式，这里可以使用apply函数：defapply_dropout(m):iftype(m)==nn.Dropout:m.train()下面是完整demo代码：#coding:utf-8import... 继续阅读 >

2021
07-09

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

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