tensorflow解决batch_normalization模型调用遇到的问题和模型引入正则项的问题

本文解决了一个在调用训练好的Mobilenet_v2模型时出现的输出结果不准确的问题,通过调整batch_normalization层的train参数设置,并增加更新操作,成功解决了模型输出错误的情况。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

最近遇到一个比较奇怪的问题,就是调用训练好的mobilenet_v2时出现模型输出结果全是瞎猜的!最后发现batch_normalize的问题

训练过程中将tf.layers.batch_normalization中的train设置成True,验证集设置成False.发现验证集的准确率保持不变了,如果将验证集也设置成True,验证准确率会提升,但是在模型调用的时候就会出现上述输出全是瞎猜的。需要增加一个更新的操作;

Update = tf.get_collection(tf.GraphKeys.UPDATE_OPS)
with tf.control_dependencies(Update):
    train_op = optimizer.minimize(loss)

然后我们就能够最小化损失函数进行训练了,之后我们将训练设置成True 验证设置成False,就不会出现任何问题了。

还有一个就是正则化中之前用sess=tf.InteractiveSession()创建会话后报了一个cpu和Gpu节点不匹配的问题!改用sess=tf.Session()后就正常了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值