Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

About code test. #5

Open
JT-Sun opened this issue Apr 26, 2023 · 3 comments
Open

About code test. #5

JT-Sun opened this issue Apr 26, 2023 · 3 comments

Comments

@JT-Sun
Copy link

JT-Sun commented Apr 26, 2023

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!

@ChanglongJiangGit
Copy link
Owner

Thanks for your attention. For the first question, maybe you can change the <your_gpu_ids> when running the script 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 <test_batch_size> you set in config.py is too large. When I tested on 2080ti, the batch-size I used is 12. Maybe this can help you!

@JT-Sun
Copy link
Author

JT-Sun commented May 5, 2023

Thanks for your reply! Thanks for your great work!

@ChanglongJiangGit
Copy link
Owner

No ploblem~

# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants