ExplainableML / czsl

PyTorch CZSL framework containing GQA, the open-world setting, and the CGE and CompCos methods.
GNU General Public License v3.0
111 stars 27 forks source link

use C-GQA datasets will make cuda memory be used up quickly #39

Open siminzheng opened 2 months ago

siminzheng commented 2 months ago

I found that when I use C-GQA datasets, I will get this problem "RuntimeError: CUDA out of memory" . The Navida gpu momery will be used up even though I use RTX 4090 (24GB) or V-100(32GB) . However , the problem won't happen when I use mit-states and ut-zappos datasets. I don't know if any processed steps exist before using C-GQA datasets。I can't solve it, Please help me.