net.train():模型训练时的特征。
net.eval():测试时的网络特征。
注:1. 模型训练时,需要使用dropout的方法,丢掉一半的参数,防止模型过拟合,而在测试时,所有的模型参数都必须保留。
2. 模型训练时,模型的输入单位是min_batch,模型测试时,是一张图片。
参考链接:
https://blog.csdn.net/edward_zcl/article/details/101947941
https://blog.csdn.net/u011276025/article/details/78507950
Pytorch中的net.train 和 net.eval使用
最新推荐文章于 2022-03-27 22:14:37 发布