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

adding 'with_cp' option to beit #1528

Closed
ydhongHIT opened this issue Apr 28, 2022 · 6 comments
Closed

adding 'with_cp' option to beit #1528

ydhongHIT opened this issue Apr 28, 2022 · 6 comments
Assignees

Comments

@ydhongHIT
Copy link

The memory cost of beit backbone is huge and you didn't implement the 'with_cp' option to save the memory. However, I add it by myself and the gpu memory doesn't get smaller. Maybe my implement has some problem or there is any reason for this phenomenon?

@MengzhangLI
Copy link
Contributor

MengzhangLI commented Apr 28, 2022

Juse because the memory on readme is set cudnn_benchmark = False. You could modify it and try again.

By the way, I really want to support DDRNet in MMSegmentation but do not have too much time to do it, would you mind helping us making a PR to support it? Thanks in advance.

@MengzhangLI MengzhangLI self-assigned this Apr 28, 2022
@ydhongHIT
Copy link
Author

Juse because the memory on readme is set cudnn_benchmark = False. You could modify it and try again.

By the way, I really want to support DDRNet in MMSegmentation but do not have too much time to do it, would you mind helping us making a PR to support it? Thanks in advance.

Thanks for your interest. I am working on another program currently and will try to PR after finishing this. The 'cudnn_benchmark' has been set to 'True'. And setting 'with_cp' to 'True' indeed reduces the memory of some models under the same settings.

@ydhongHIT
Copy link
Author

Setting 'cuda_benchmark' to 'False' helps, thank you! But setting 'with_cp' to 'True' indeed reduces the memory of some models even though 'cuda_benchmark' is 'True'.

@MengzhangLI
Copy link
Contributor

OK, thats cool! Do you have clear time schedule about your current program?

@ydhongHIT
Copy link
Author

OK, thats cool! Do you have clear time schedule about your current program?

Maybe two or three months but not sure.

@MeowZheng
Copy link
Collaborator

Juse because the memory on readme is set cudnn_benchmark = False. You could modify it and try again.
By the way, I really want to support DDRNet in MMSegmentation but do not have too much time to do it, would you mind helping us making a PR to support it? Thanks in advance.

Thanks for your interest. I am working on another program currently and will try to PR after finishing this. The 'cudnn_benchmark' has been set to 'True'. And setting 'with_cp' to 'True' indeed reduces the memory of some models under the same settings.

Thanks for suggestion, and we will add with_cp in BEiT, or would you like to create a nice pr to add it. If you have any implement problem, you might refer this pr #1431

wjkim81 pushed a commit to wjkim81/mmsegmentation that referenced this issue Dec 3, 2023
* reorganize config folder structure

* add runtime and schedule configs
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