训练时间参考
HarrisonWu42 opened this issue · 1 comments
HarrisonWu42 commented
请问不同型号的跟踪器在什么配置的机器上训练的,训练时间大概是多久
dawnyc commented
Training time for different types of models:
- ROMTrack
- stage1: 28h (4 V100 GPUs, 300 epochs)
- stage2: 12h (4 V100 GPUs, 100 epochs)
- ROMTrack (only GOT-10k)
- stage1: 15h (4 V100 GPUs, 100 epochs)
- stage2: 10h (4 V100 GPUs, 50 epochs)
- ROMTrack-384
- stage1: 24h (8 V100 GPUs, 300 epochs)
- stage2: 15h (8 V100 GPUs, 100 epochs)
- ROMTrack-384 (only GOT-10k)
- stage1: 11h (8 V100 GPUs, 100 epochs)
- stage2: 10h (8 V100 GPUs, 50 epochs)
- ROMTrack-Small-256
- stage1: 13h (8 V100 GPUs, 300 epochs)
- stage2: 6h (8 V100 GPUs, 100 epochs)
- ROMTrack-Tiny-256
- stage1: 13h (8 V100 GPUs, 300 epochs)
- stage2: 6h (8 V100 GPUs, 100 epochs)
By using A6000 GPUs, you will speed up the training process of 2 ~ 2.5 times.