Pytorch中model.train()和model.eval()
@[TOC]Crystal的博客
Pytorch中model.train()和model.eval()
问题:
刚开始接触pytorch时,发现别人的代码中,会在训练模型的一开始写上model.train(),对应的,在测试模型一开始写上model.eval()。我尝试不使用这两句,发现程序仍然能够正常运行,所以就非常好奇这两句有什么作用,为什么要这么写。
解答:
(1) 用法:这两个方法是针对在模型训练和评估时采用不同的方式的情况。如果模型中有BN层(Batch Normalization)和正则化Dro
原创
2020-10-05 16:04:00 ·
309 阅读 ·
2 评论