CAT-Net icon indicating copy to clipboard operation
CAT-Net copied to clipboard

Published 20 hours ago • verified publisher icon mjkwon2021

About training time

Open zyqss opened this issue 2 years ago • 1 comments

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

zyqss avatar Aug 23 '22 07:08 zyqss

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!

CauchyComplete avatar Aug 23 '22 07:08 CauchyComplete