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

Training Cost #21

Open
AmberCheng opened this issue Aug 23, 2024 · 1 comment
Open

Training Cost #21

AmberCheng opened this issue Aug 23, 2024 · 1 comment

Comments

@AmberCheng
Copy link

Hi, I am currently also fine-tuning the motion module of Animatediff. However, I am using the original GitHub(https://github.com/guoyww/AnimateDiff). I found that when the training size is 512x512x16 frames, the GPU memory has reached 51G. I wonder what are your training configurations and resource consumption? Looking forward to your reply~

@tumurzakov
Copy link
Owner

Thats too many, 512x512x16 must took somewhere 15-16gb, I guess you didn't turn off gradients on unet. You must turn on grandients only on motion module.

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