Gradient Centralization TensorFlow:实现Gradient Centralization并允许其在TensorFlow中用作Py

梯度集中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