search

Kroery / DiffMOT

code for CVPR2024 paper: DiffMOT: A Real-time Diffusion-based Multiple Object Tracker with Non-linear Prediction
MIT License
379 stars 42 forks source link

Training time #3

Closed dyhBUPT closed 6 months ago

dyhBUPT commented 6 months ago

Hi, thanks for your excellent work!

I'm interested in the training cost of DiffMOT. I see that you train DiffMOT on 4 3090 GPUs with batch size 2048 for 800 epochs. Could you please tell us how much time does it spend?

In comparison, DiffusionTrack (AAAI2024) is trained on 8 3090 GPUs for 30 hours.

Thanks!

Kroery commented 6 months ago

DiffMOT is trained on 4 3090 GPUs for about 6 hours on DanceTrack.

dyhBUPT commented 6 months ago

Thanks~