文件名称:Gradient-Centralization-TensorFlow:实现Gradient Centralization并允许其在TensorFlow中用作Python包
文件大小:224KB
文件格式:ZIP
更新时间:2024-03-24 02:32:51
python machine-learning deep-learning neural-network tensorflow
梯度集中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
【文件预览】:
Gradient-Centralization-TensorFlow-main
----MANIFEST.in(44B)
----.github()
--------PULL_REQUEST_TEMPLATE.md(579B)
--------ISSUE_TEMPLATE()
--------workflows()
----images()
--------gctf.png(98KB)
--------projected_grad.png(87KB)
--------illutstration.png(37KB)
----examples()
--------gctf_mnist.ipynb(20KB)
--------gctf_horses_v_humans.ipynb(28KB)
----LICENSE(11KB)
----setup.py(1KB)
----README.md(6KB)
----gctf()
--------__init__.py(192B)
--------centralized_gradients.py(3KB)
--------optimizers.py(3KB)
--------version.py(22B)
----CODE_OF_CONDUCT.md(3KB)
----.gitignore(2KB)