search

Alexadlu / AFter

Attention-based Fusion Router for RGBT Tracking
2 stars 0 forks source link

single GPU #5

Open yunqi831017 opened 2 weeks ago

yunqi831017 commented 2 weeks ago

Hello, may I ask how to set up if I want to train with a single GPU

hello-wwy commented 1 week ago

You could set multi_gpu to False in train_settings/tomp/tomp_v1.py and specify gpu through CUDA_VISIBLE_DEVICES. Then, you can train with a single gpu by running run_training.py.