星星博客 »  > 

pytorch model.train() 开启batchnormalize 和 dropout model.eval() 则会关闭dropout

https://blog.csdn.net/kking_edc/article/details/104663305

相关文章