bytedance / ColTrack

This repository is an official implementation of Collaborative Tracking Learning for Frame-Rate-Insensitive Multi-Object Tracking
Apache License 2.0
21 stars 1 forks source link

训练成本提问 #1

Closed mmm-cc closed 10 months ago

mmm-cc commented 10 months ago

请问在mot17和dancetrack上训练一次,分别要在什么gpu下训练多久啊

mmm-cc commented 10 months ago

python test.py可以多卡吗,一块A800好像显存也不够啊

easonbyte commented 10 months ago

我是使用8张V100训练,第一阶段很快,一两天能完成。第二阶段比较慢,mot17需要五六天的样子,Dancetrack要一周多。训练和测试都可以多卡,测试脚本可以参考code

mmm-cc commented 10 months ago

好的好的,非常感谢您的回答