search

mjkwon2021 / CAT-Net

Official code for CAT-Net: Compression Artifact Tracing Network. Image manipulation detection and localization.
222 stars 26 forks source link

About training time #28

Open zyqss opened 2 years ago

zyqss commented 2 years ago

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 commented 2 years ago

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!