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

memory cost #42

Open
VLadImirluren opened this issue Sep 3, 2023 · 2 comments
Open

memory cost #42

VLadImirluren opened this issue Sep 3, 2023 · 2 comments

Comments

@VLadImirluren
Copy link

I haven't as large compute resource as yours. This make me hard to reproduce your work.
Can you share some trick of reduce the memory cost but reach the same effect?
(In brief, how to use lower batchsize but same result)

@ARuiChen
Copy link
Collaborator

ARuiChen commented Sep 3, 2023

1、Installing xformers can greatly reduce memory usage.
2、Use gradient accumulation technology that is easy to implement as illustrated in threestudio
3、https://github.com/threestudio-project/threestudio/blob/main/README.md?plain=1#L575
4、rent
5、I think 4 GPUs are enough to reproduce my work, but there should be some changes needed in the configuration. For geometry modeling, use the DMTet 128 resolution to achieve a balance between quality and memory consumption when limited to batch size. For appearance modeling, the addition of negative prompt can greatly improve the generation quality.

@VLadImirluren
Copy link
Author

1、Installing xformers can greatly reduce memory usage. 2、Use gradient accumulation technology that is easy to implement as illustrated in threestudio 3、https://github.com/threestudio-project/threestudio/blob/main/README.md?plain=1#L575 4、rent 5、I think 4 GPUs are enough to reproduce my work, but there should be some changes needed in the configuration. For geometry modeling, use the DMTet 128 resolution to achieve a balance between quality and memory consumption when limited to batch size. For appearance modeling, the addition of negative prompt can greatly improve the generation quality.

thanks

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