用于Tensorflow的PID优化器
PID优化的Tensorflow实现(CVPR 2018)。
有关算法的详细信息,请参见以下论文。
该程序已在Tensorflow r1.4上进行了测试。
安装
$ git clone https://github.com/machida-mn/tensorflow-pid
$ cd tensorflow-pid
$ pip install .
用法
import tensorflow_pid
loss = .......
train_op = tensorflow_pid . PIDOptimizer ( learning_rate = 0.01 , kd = 0.001 ). minimize ( loss )
在其他框架中实施
火炬
2021-12-12 12:20:46
4KB
Python
1