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

GPU #5

Open
MeycL opened this issue Mar 6, 2024 · 5 comments
Open

GPU #5

MeycL opened this issue Mar 6, 2024 · 5 comments

Comments

@MeycL
Copy link

MeycL commented Mar 6, 2024

What GPU do you use for experiments? How much VRAM is required approximately? If I want to use a 3090, how should I adjust the parameters to make it work?

@wangxiang1230
Copy link
Contributor

What GPU do you use for experiments? How much VRAM is required approximately? If I want to use a 3090, how should I adjust the parameters to make it work?

Hi, our experiments are conducted on 4 V100 (32G) or 4 A100 (80G) GPUs. For 3090 GPU, you can change "QUERY_PER_CLASS: 5" to "QUERY_PER_CLASS: 2". Thanks.

@MeycL
Copy link
Author

MeycL commented Mar 7, 2024

Thank you very much for your prompt response. I will try your suggestion.

@ssp789
Copy link

ssp789 commented Jun 15, 2024

Hello, how long does it take for 3090 to train the largest dataset with 10 epochs

@wangxiang1230
Copy link
Contributor

Hello, how long does it take for 3090 to train the largest dataset with 10 epochs

Hi, maybe within one day.

@ssp789
Copy link

ssp789 commented Jun 15, 2024

Hello, how long does it take for 3090 to train the largest dataset with 10 epochs

Hi, maybe within one day.

Thank you very much!

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

No branches or pull requests

3 participants