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

What are the compute resources required for this? #3

Open
yxchng opened this issue Nov 9, 2023 · 5 comments
Open

What are the compute resources required for this? #3

yxchng opened this issue Nov 9, 2023 · 5 comments

Comments

@yxchng
Copy link

yxchng commented Nov 9, 2023

Can this be trained on V100 32gb or A100 40gb? The paper mentioned A100 but doesn't mentioned whether it is 40gb or 80gb.

@wusize
Copy link
Owner

wusize commented Nov 9, 2023

We use A100 80g. But A100 40g or V100 can also be enough for training EVA-CLIP. Just set batch size to 1 in case of "out of memory".

@cilinyan
Copy link

cilinyan commented Feb 6, 2024

Could you please provide the estimated training time for CLIPSelf on the COCO and CC3M datasets?
Thanks!

@wusize
Copy link
Owner

wusize commented Feb 7, 2024

The training of EVA-CLIP models is fast. With 8 A100-80Gs, it takes about 2 hours to train a ViT-B/16 on COCO for 6 epochs costs and 6 hours for ViT-L/14.

@cilinyan
Copy link

cilinyan commented Feb 7, 2024

Thank you for your reply!

The training of EVA-CLIP models is fast. With 8 A100-80Gs, it takes about 2 hours to train a ViT-B/16 on COCO for 6 epochs costs and 6 hours for ViT-L/14.

@kunpeng337
Copy link

We use A100 80g. But A100 40g or V100 can also be enough for training EVA-CLIP. Just set batch size to 1 in case of "out of memory".

Hello, you mentioned here that "A100 40g or V100 is also enough to train EVA-CLIP", I want to ask is it also referring to 8 pieces of A100 40g or 8 pieces of V100? Since our lab has two pieces of 3090, I would also like to ask what is the minimum configuration required for open vocabulary object detection in the paper?

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

4 participants