search

hustvl / SparseTrack

Official PyTorch implementation of SparseTrack (the new version of code will come soon)
MIT License
133 stars 12 forks source link

Single card training #5

Closed yangwhf closed 9 months ago

yangwhf commented 1 year ago

Hello. Thank your for sharing the wonderful works. Your repository has been an invaluable resource for me. Can the model be trained using a single card? How to implement it?

Ali96528 commented 1 year ago

Setting the option "is_distributed = False" in train_config allows for single-card training. Don't forget to modify batchsize and num_of_workers.