net.train()
在训练模型时需要在前面加上。
net.eval()
在非训练的时候是需要加的,没有这句代码,一些网络层的值会发生变动,不会固定,神经网络每一次生成的结果也是不固定的,如droupout层和BN层,生成质量可能好也可能不好。
训练阶段net.train()&&测试阶段net.eval()的作用
最新推荐文章于 2023-11-29 16:26:57 发布
net.train()
在训练模型时需要在前面加上。
net.eval()
在非训练的时候是需要加的,没有这句代码,一些网络层的值会发生变动,不会固定,神经网络每一次生成的结果也是不固定的,如droupout层和BN层,生成质量可能好也可能不好。