Skip to content
New issue

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

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

Already on GitHub? Sign in 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~

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

No branches or pull requests

2 participants