search

open-mmlab / mmdetection

OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io
Apache License 2.0
29.08k stars 9.38k forks source link

CenterNet training extremely slow #10249

Open MLRadfys opened 1 year ago

MLRadfys commented 1 year ago

Hi,

Iam trying to train CenterNet on the Coco dataset using the _centernet-update_r50_fpn_8xb8-amp-lsj-200ecoco.py config. According to the log output, training will take about 25 days. This seems super long! Now Iam wondering if there is anything that I could check in order to speed up training?

Thanks in advance,

kind regards,

Michael

hhaAndroid commented 1 year ago

@MLRadfys Because of the training 200e, you may consider scaling down the epoch if it is a custom dataset