Stark
Stark copied to clipboard
在got10k数据集上进行测试出现:out of memory
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
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>"
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.