Skip to content
New issue

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

#

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

Already on GitHub? # to your account

Jump to bottom

How to improve this model's train effeciency on GPU #4

Open
haonanSUniv opened this issue Apr 15, 2023 · 0 comments
Open

How to improve this model's train effeciency on GPU #4

haonanSUniv opened this issue Apr 15, 2023 · 0 comments
Labels
help wanted Extra attention is needed

Comments

@haonanSUniv
Copy link

I tried to run the part of cpcv of this model on Google Colab Pro+ with A100 GPU but it only uses 1.7GB over 40GB GPU RAM. No matter how I changed the batch size, worker_num of the GPU, thread_num of the CPU. It seems to me that there is a bottleneck that restrains the GPU performance and train speed. Can anyone help with this problem?
@YangletLiu YangletLiu added the help wanted Extra attention is needed label Apr 22, 2023
# for free to join this conversation on GitHub. Already have an account? # to comment
Assignees
No one assigned
Labels
help wanted Extra attention is needed
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@YangletLiu @haonanSUniv