Skip to content

add width_mult to optimizer dict#65

Open
marcobellagente93 wants to merge 1 commit intomicrosoft:mainfrom
marcobellagente93:add-with-optimizer-dict
Open

add width_mult to optimizer dict#65
marcobellagente93 wants to merge 1 commit intomicrosoft:mainfrom
marcobellagente93:add-with-optimizer-dict

Conversation

@marcobellagente93
Copy link

Adding the width_mult key to the MuAdam state dictionary to make it more easy to use the class, e.g. to enable its correct use in https://github.com/EleutherAI/gpt-neox

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant

Comments