Skip to content

Conjugate Gradient with Quadratic Line Search

Notifications You must be signed in to change notification settings

zh272/CGQ-solver

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CGQ: Conjugate Gradient with Quadratic Line Search [ECMLPKDD2021]

@inproceedings{
  hao2021cgq,
  title={Adaptive Learning Rate and Momentum for Training Deep Neural Networks},
  author={Zhiyong Hao and Yixuan Jiang and Huihua Yu and Hsiao-Dong Chiang},
  booktitle={ECMLPKDD},
  year={2021},
}

Dependencies:

numpy, pytorch, matplotlib, sklearn,scipy

Example usage:

# Compare optimizer performances on same initialization point:
python train_net.py --sgd_epochs=200 --ds_name=cifar10 --batch_size=64 --gpu_id=0 --model=resnet --figure_suffix='test_new/' --seed=6022

About

Conjugate Gradient with Quadratic Line Search

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages