Skip to content

Feat/consolidate optimizers#11

Open
alexander-darwiche wants to merge 12 commits into
feature/adamfrom
feat/consolidate-optimizers
Open

Feat/consolidate optimizers#11
alexander-darwiche wants to merge 12 commits into
feature/adamfrom
feat/consolidate-optimizers

Conversation

@alexander-darwiche

Copy link
Copy Markdown

Adding functionality to run all 3 types of optimizer that are currently implemented from the same repository. Added in the necessary arguments for each optimizer and a new argument to specify the desired optimizer to be used.

alexander-darwiche and others added 4 commits October 31, 2024 18:00
* Created Lion Optimizer

Created a private version of Lion Optimizer

* Updated trainer.py

Added DP-Lion to the choose_optimizer along with GradSampleModule which is required for per sample grad calculations
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants