GPU out of memory when testing on composite-1k

Yaoyi-Li / GCA-Matting

Official repository for Natural Image Matting via Guided Contextual Attention

MIT License

392 stars 62 forks source link

GPU out of memory when testing on composite-1k #5

Closed changliu19 closed 4 years ago

changliu19 commented 4 years ago

Hi Yaoyi,

Thank you for your remarkable work!

I noticed in the README it says that 8GB is needed when testing on conposite-1k dataset. However, when I run tests with a GPU with 11GB memory (RTX2080Ti), the program raised an out of memory error. So I want to ask that do I need to resize the image first before running or change the 'scale' option in the toml file? Thank you!

My testing dataset is generated by the Composition_code.py provided with the Adobe Image Matting dataset.

changliu19 commented 4 years ago

Issue closed since I have resolved the problem. Thanks!

umiiwa commented 4 years ago

Issue closed since I have resolved the problem. Thanks!

Could you please tell me how you resolved the problem? I've encountered the same problem.