Open JT-Sun opened 1 year ago
Thanks for your attention. For the first question, maybe you can change the python test.py --gpu <your_gpu_ids>
. For example, if you want to use gpu 0,1,2,3, you can change it to 0,1,2,3
or 0-3
. For the second question, this kind of problem is usually caused by the config.py
is too large. When I tested on 2080ti, the batch-size I used is 12. Maybe this can help you!
Thanks for your reply! Thanks for your great work!
No ploblem~
Hello, I tried the installation instructions to test your code, but no matter how many Gpus I set up it will only run on Gpus 0. Also when I use one 3090 as gpu, it said cuda out of memory. But your paper said you test on a single 2080ti gpu. Looking forward to your reply!