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

train not Stably maintain high usage #47

Open
LiaoRiYang opened this issue Mar 7, 2024 · 1 comment
Open

train not Stably maintain high usage #47

LiaoRiYang opened this issue Mar 7, 2024 · 1 comment

Comments

@LiaoRiYang
Copy link

when i train my dataset, GPU not Stably maintain high usage
(it sometimes 10% , and sonetimes 70%,The numbers will go up and down)
batch size 8 (because RTX3060 only has 12 GPU RAM)
my num work is set 32, because i have 32 core CPU
threads is set 1 , because only has one GPU
Do any parameters still need to be changed?

@LSH9832
Copy link
Owner

LSH9832 commented Mar 7, 2024

just set num_workers=4. Too large number of num_workers may cause the opposite effect.

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

2 participants