第一句子网 - 唯美句子、句子迷、好句子大全
第一句子网 > model.train() model.eval()

model.train() model.eval()

时间:2022-02-20 15:49:42

相关推荐

model.train()  model.eval()

文章目录

1、model.train2、model.eval

1、model.train

model.train()是在模型训练的时候使用,主要针对Batch Normalization 和 Dropout 方法模式,防止网络过拟合。

2、model.eval

model.eval()是在模型检验的时候使用的,也是针对BN与Dropout方法模式的,pytorch会自动把BN和DropOut固定住,不会取平均,而是用训练好的值。具体目的大概是为了防止图片失真。

因此,在使用Pytorch训练和评价模式的时候,一定要把实例化的model指定为对应的模式。

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。