Hi, what kind of Graphics card and how much the card's memory is required for training?

CaptainEven / MCMOT

Real time one-stage multi-class & multi-object tracking based on anchor-free detection and ReID

MIT License

383 stars 82 forks source link

Hi, what kind of Graphics card and how much the card's memory is required for training? #59

Closed ShellyLingling closed 3 years ago

CaptainEven commented 3 years ago

@ShellyLingling It depends on the backbone you use, for Resnet18 backbone, 1060 will be good enough, but for DCN， 8G memory at least i think.

ShellyLingling commented 3 years ago

Thank you!! May I ask how it works with Resnet18 backbone?

CaptainEven commented 3 years ago

@ShellyLingling Of course， arch resdcn_18 in opts.py is resnet18(Although it's named resdcn, actually no dcn used, you can find it in network definition).