Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Gemini] more tests for Gemini #2038

Merged
merged 2 commits into from
Nov 29, 2022
Merged

[Gemini] more tests for Gemini #2038

merged 2 commits into from
Nov 29, 2022

Conversation

feifeibear
Copy link
Contributor

What's this PR for

add more tests for the optimizer test of Gemini. I find that test_optim.py can not work for simple_net! You can merge this PR. We should fix the simple_net bug in another PR.

optimizer.backward(loss)
return logits
# 'gpt2', 'bert',
TEST_MODELS = ['gpt2', 'bert']
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

can not work for simple_net! @1SAA

@1SAA 1SAA merged commit eb7742a into hpcaitech:main Nov 29, 2022
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.

None yet

2 participants