-
Notifications
You must be signed in to change notification settings - Fork 414
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
Hardware and data requirements for model training. #115
Comments
I would also like to know. I attempted training with an RTX 3060 and batch size of 1 and ran out of memory. I have 12 GB of dedicated GPU memory and 16 GB of shared GPU memory.
|
I have stage 1 training running with my RTX 3060 and 64 GB of RAM. That's 12 GB of GPU memory and 32 GB of shared GPU memory totaling 44 GB of GPU memory. It seems to be just barely enough. |
Closed
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
What are the GPU requirements to run the training and approximately how many input videos should be used for training?
The text was updated successfully, but these errors were encountered: