tensorflow中滑动平均值的用法与注意事项

本文通过实例演示了如何在TensorFlow中使用滑动平均值来稳定模型训练过程。详细介绍了ExponentialMovingAverage类的使用,包括衰减率(decay)的设置及其对模型的影响。通过代码展示了影子变量的创建和更新过程,以及如何计算变量的滑动平均值。

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

import tensorflow as tf

x = tf.Variable(0.0)

#定义一个滑动平均值类。decay越大模型越稳,一般设成非常接近1的数(0.99或0.999)
#如果设置了num_updata参数,那么该滑动平均值会随着更新速度的变化而变化
averages_class = tf.train.ExponentialMovingAverage(0.99)

#为变量列表中的每个变量创建一个影子变量。
#tf.trainable_variables()返回集合图上中,没有指定trainable_variables= False的变量
average_op = averages_class.apply(tf.trainable_variables())

#计算变量的滑动平均值
average = averages_class.average(x)

a = tf.global_variables_initializer()

with tf.Session() as sess:

    sess.run(a)

    for i in range(6):
        
        #每次循环都要更新影子变量的值
        sess.run(average_op)
		
        print(sess.run(x),"    ",sess.run(average))
					#0.0          0.0
					#5.0          0.049999952
					#10.0         0.14949986
					#15.0         0.29800472
					#20.0         0.4950245
					#25.0         0.74007404


        #每次循环x加5
        y = tf.add(x, 5.0)
		
		#重新给x赋值,并更新
        sess.run(tf.assign(x, y))
        

decay(衰减率)的选择,对模型的影响
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值