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.

Sign up for GitHub

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

Jump to bottom

Add model training GPU regression test #10881

Open
EtiennePerot opened this issue Sep 9, 2024 · 0 comments
Open

Add model training GPU regression test #10881

EtiennePerot opened this issue Sep 9, 2024 · 0 comments
Labels
area: gpu Issue related to sandboxed GPU access type: enhancement New feature or request

Comments

@EtiennePerot
Copy link
Contributor

Description

Our only GPU regression tests are doing inference. We should have a test that does some training too.

Is this feature related to a specific bug?

#10879

Do you have a specific solution in mind?

#10879 has an example Dockerfile which is modeled after karpathy/llm.c#677.
@EtiennePerot EtiennePerot added the type: enhancement New feature or request label Sep 9, 2024
@ayushr2 ayushr2 added the area: gpu Issue related to sandboxed GPU access label Oct 22, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
area: gpu Issue related to sandboxed GPU access type: enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@EtiennePerot @ayushr2