Add optional momentum acceleration to lstsq_grad object updates.
Approach
Computes the ADAM update from the lstsq_grad object update. Only adds ADAM to the object update and not the probe update because probe updates are unstable when accelerated.
Pre-Merge Checklists
Submitter
[ ] Write a helpfully descriptive pull request title.
[ ] Organize changes into logically grouped commits with descriptive commit messages.
[ ] Document all new functions.
[ ] Click 'details' on the readthedocs check to view the updated docs.
[ ] Write tests for new functions or explain why they are not needed.
[ ] Address any complaints from pep8speaks.
Reviewer
[ ] Actually read all of the code.
[ ] Run the new code yourself; the included tests should make this easy.
[ ] Write a summary of the changes as you understand them.
carterbox
commented
1 year ago
Purpose
Add optional momentum acceleration to lstsq_grad object updates.
Approach
Computes the ADAM update from the lstsq_grad object update. Only adds ADAM to the object update and not the probe update because probe updates are unstable when accelerated.
Pre-Merge Checklists
Submitter
Reviewer