Tested under python2.
- python packages
- matplotlib==2.0.2
python mnist_pid.py
python mnist_momentum.py
python compare.py
If PIDOptimizer is used in your paper/experiments, please cite the following papers.
@inproceedings{pid2018,
title={A PID Controller Approach for Stochastic Optimization of Deep Networks},
author={Wangpeng An and Haoqian Wang and Qingyun Sun and Jun Xu and Qionghai Dai and Lei Zhang},
booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
year={2018}
}