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

About training time #28

Open
zyqss opened this issue Aug 23, 2022 · 1 comment
Open

About training time #28

zyqss opened this issue Aug 23, 2022 · 1 comment

Comments

@zyqss
Copy link

zyqss commented Aug 23, 2022

Very happy to see your work!
When I was downloading the tampcoco dataset, I found 40G of content and would like to ask you how long is the training duration in 2X NVIDIA TITAN RTX?
If I have only one NVIDIA TITAN RTX, will it lead to a long trial period?
Thank you for answering my questions

@CauchyComplete
Copy link
Collaborator

Hi,
Actually, tampCOCO is unnecessarily large for training CAT-Net.
You might find out that one epoch does not use all training images. Instead, the maximum number of images used in training for each dataset is limited to about 1k due to balanced sampling.
As far as I remember, training took about 1~2 days for 200 epochs when using two TITAN RTX.
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