深度学习中train()和eval()的作用

颜娘娘的碎碎念

 关注

阅读 77

2023-07-14

（1）在训练时要加上train()。在测试和使用时要加上eval（)。它们两个函数用来控制【batchNorm层，dropout层等用于优化训练而添加的网络层】的开关。

Train()会开启那些层，eval会关闭那些层。

（2）dropout层的作用，在每一次的训练过程中，随机舍弃一定比例的神经元，每次训练舍弃的神经元都不同，可以有效防止模型在训练过程中过拟合。

相关推荐

深度学习中momentum的作用

陌岛 82 0 0

Pytorch：model.train()和model.eval()用法和区别，以及model.eval()和torch.no_grad()的区别

史值拥 37 0 0

pytorch model.train() 和model.eval() 对 BN 层的影响

清冷的蓝天天 224 0 0

model.train()和model.eval()的用法及model.eval()可能导致测试准确率的下降

惠特曼 151 0 0

动手学深度学习09.Softmax——Accuracy()、train()

扒皮狼 84 0 0

深度学习中的深度学习

幺幺零 74 0 0

Python中eval如何使用？其作用是什么？

萧让听雪 85 0 0

【深度学习】——深度学习中的梯度计算

蒸熟的土豆 279 0 0

lambda和eval()

weipeng2k 80 0 0

Ajax中的eval函数的用法

月白色的大狒 59 0 0

精彩评论（0）

0 0 举报