训练好的模型,对同一个数据进行predict,发现每次结果都不一样。
原因是在model.eval()之前调用了torchsummary.summary来查看模型结构,说明summary函数会改变模型权重(只改变BN还是都改变?),把summary函数放到model.eval()之后发现输出就保持不变,说明summary函数会改变model的BN参数(我的model里没有dropout)。
net.eval() # freeze BN and Dropout summary(net, input_size=(1, 24, 200, 200))