This repository contains source code of the paper:
- 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 paper.
@InProceedings{An_2018_CVPR,
author = {An, Wangpeng and Wang, Haoqian and Sun, Qingyun and Xu, Jun and Dai, Qionghai and Zhang, Lei},
title = {A PID Controller Approach for Stochastic Optimization of Deep Networks},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2018}
}