Stark icon indicating copy to clipboard operation
Stark copied to clipboard

在got10k数据集上进行测试出现:out of memory

Open hongsheng-Z opened this issue 3 years ago • 2 comments

Hello when I use python tracking/test.py stark_st baseline_got10k_only --dataset got10k_test --threads 32 python lib/test/utils/transform_got10k.py --tracker_name stark_st --cfg_name baseline_got10k_only Tested on the got10k data set, the result is less than 20 Sequences, and the oom error occurs. My hardware platform is 3090

hongsheng-Z avatar Sep 20 '21 05:09 hongsheng-Z

Hi, test.py by default assumes that the user is using 8 GPUs for evaluation and distributes threads evenly to each GPU. You can specify the number of threads and the number of GPUs by using "python tracking/test.py stark_st baseline_got10k_only --dataset got10k_test --threads <NUM_THREADS> --num_gpus <NUM_GPUS>"

MasterBin-IIAU avatar Sep 22 '21 09:09 MasterBin-IIAU

Okay, thank you for your reply. I set the number of threads to 32 before, and then I changed it to 12 and started working. But currently, the FPS on GOT10k is only about 6.

hongsheng-Z avatar Sep 22 '21 12:09 hongsheng-Z