训练时间参考

[HarrisonWu42]

请问不同型号的跟踪器在什么配置的机器上训练的，训练时间大概是多久

[dawnyc]

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.