上传者: 42140710
|
上传时间: 2021-10-23 17:29:24
|
文件大小: 224KB
|
文件类型: -
梯度集中TensorFlow
这个Python套件在TensorFlow中实现了梯度集中,这是Yong等人建议的一种简单有效的针对深度神经网络的优化技术。 。 它既可以加速训练过程,又可以提高DNN的最终泛化性能。
安装
运行以下命令进行安装:
pip install gradient-centralization-tf
用法
为指定的优化器创建集中式渐变函数。
参数:
optimizer :一个tf.keras.optimizers.Optimizer object 。 您正在使用的优化程序。
例子:
>> > opt = tf . keras . optimizers . Adam ( learning_rate = 0.1 )
>> > optimizer . get_gradients = gctf . centralized_gradients_for_optimize