akshitac8 / Generative_MLZSL

[TPAMI 2023] Generative Multi-Label Zero-Shot Learning
https://akshitac8.github.io/GAN_MLZSL/
GNU General Public License v3.0
50 stars 14 forks source link

How much gpu memory does it costs? #7

Closed Neko9810 closed 2 years ago

Neko9810 commented 2 years ago

I run train_nus_wide on a single 1080Ti, but it returns CUDA out of memory.

akshitac8 commented 2 years ago

@bk820653874 thank you for your interests in our work. Can you share the details of your GPU card?